add requirements.txt (#287)

* add requirements.txt Signed-off-by: Yimi81 <[email protected]> * add protobuf package --------- Signed-off-by: Yimi81 <[email protected]>
01-ai · Jan 11, 2024 · 88b731e · 88b731e
1 parent e1a875a
commit 88b731e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,8 @@
+transformers>=4.36.2
+gradio>=4.13.0
+protobuf>=4.25.1
+torch==2.0.1
+accelerate
+sentencepiece
+deepspeed
+datasets