浏览代码

convert-hf-to-gguf : require einops for InternLM2ForCausalLM (#5792)

nold 1 年之前
父节点
当前提交
da3b9ba2b7
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1 0
      requirements/requirements-convert-hf-to-gguf.txt

+ 1 - 0
requirements/requirements-convert-hf-to-gguf.txt

@@ -1,2 +1,3 @@
 -r ./requirements-convert.txt
 torch~=2.1.1
+einops~=0.7.0