update readme

X-D-Lab · Sep 28, 2023 · f4c0459 · f4c0459
1 parent 9a35d8f
commit f4c0459
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -5,6 +5,7 @@
 ```
 from transformers import AutoModelForCausalLM, AutoTokenizer
 from transformers.generation import GenerationConfig
+from modelscope import snapshot_download
 
 model_dir = snapshot_download("thomas/test", revision = 'v1.0.10')
 

diff --git a/requirements.txt b/requirements.txt
@@ -0,0 +1,6 @@
+transformers==4.32.0
+accelerate
+tiktoken
+einops
+transformers_stream_generator==0.0.4
+scipy