DeepSeek-R1-FTCD / requirements.txt
MRasheq's picture
First Commit
9716d05
raw
history blame contribute delete
319 Bytes
--extra-index-url https://download.pytorch.org/whl/cu118
numpy>=1.24.0
torch==2.1.2
transformers==4.36.2
accelerate==0.25.0
bitsandbytes==0.41.1
pandas==2.1.4
gradio==4.8.0
peft==0.7.1
datasets==2.15.0
sqlalchemy==2.0.23
psycopg2-binary==2.9.9
scipy==1.11.4
tqdm>=4.65.0
safetensors>=0.4.0
packaging>=23.0
einops>=0.6.1