MoMA_zeroGPU / requirements.txt
Kunpeng Song
fix zero
0b29f0b
raw
history blame
421 Bytes
pip
einops
fastapi
gradio
numpy
requests
sentencepiece
tokenizers>=0.12.1
torch==2.0.1
torchvision==0.15.2
uvicorn
wandb
shortuuid
httpx==0.24.0
deepspeed==0.13.1
peft==0.4.0
transformers==4.36.2
accelerate==0.21.0
bitsandbytes==0.41.0
scikit-learn==1.2.2
sentencepiece==0.1.99
einops==0.6.1
einops-exts==0.0.4
timm==0.6.13
gradio_client
opencv-python
diffusers
torchaudio
torchmetrics
llava-torch
rembg
pytorch_lightning