Yuan2-2B-demo / requirements.txt
IEIT-Yuan's picture
Add accelerate (#1)
87f8f2a verified
raw
history blame contribute delete
117 Bytes
torch==2.1.2
transformers==4.30.2
gradio>=4.13.0
sentencepiece==0.1.99
einops==0.6.1
numpy==1.22.2
accelerate==0.26.1