更新requirements

9 months ago · e3071eb677
parent 7719d75c20
commit e3071eb677
2 changed files with 6 additions and 0 deletions
--- a/.gitignore
+++ b/.gitignore
@ -7,6 +7,7 @@ build
 .vscode/
 .idea/
 .DS_Store
+.venv

 /private/
 output_qwen/
--- a/requirements.txt
+++ b/requirements.txt
@ -4,3 +4,8 @@ tiktoken
 einops
 transformers_stream_generator==0.0.4
 scipy
+auto-gptq>=0.5.1
+transformers>=4.35.0
+optimum>=1.14.0
+peft<0.8.0
+deepspeed