Merge pull request #452 from zhanglu0704/dev-zl

add requirements.txt for Dockerfile
FlagAI-Open · Aug 9, 2023 · dec374c · dec374c
2 parents 9ac9c3c + 75adf40
commit dec374c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,5 @@
+packaging
+PyYAML==5.4.1
+deepspeed==0.6.5
+flash-attn==1.0.2
+bminf