Merge pull request #91 from Gmgge/main

upgrade:transformers checks model dependency einops
01-ai · Nov 9, 2023 · 2320d7d · 2320d7d
2 parents 49cd580 + 0a9af34
commit 2320d7d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -6,3 +6,4 @@ auto-gptq==0.4.2
 datasets==2.14.5
 optimum==1.13.2
 exllama==0.1.0
+einops==0.7.0