fix(convert-hf-to-gguf): requires einops for InternLM2ForCausalLM models

2024-02-29 09:26:40 +01:00 · 2024-02-29 09:26:40 +01:00 · 0046e58d85
commit 0046e58d85
parent d5ab29757e
1 changed files with 1 additions and 0 deletions
--- a/requirements/requirements-convert-hf-to-gguf.txt
+++ b/requirements/requirements-convert-hf-to-gguf.txt
@ -1,2 +1,3 @@
 -r ./requirements-convert.txt
 torch~=2.1.1
+einops~=0.7.0