PusaV1 / requirements.txt
rahul7star's picture
Update requirements.txt
456d954 verified
raw
history blame contribute delete
756 Bytes
gradio
torch
torchvision
cupy-cuda12x ; sys_platform != "win32"
transformers==4.46.2
controlnet-aux==0.0.7
imageio
imageio[ffmpeg]
safetensors
einops
sentencepiece
protobuf
modelscope
ftfy
huggingface_hub
git+https://github.com/huggingface/diffusers.git
transformers
accelerate
safetensors
peft
torch>=2.0.0
torchvision
transformers==4.46.2
controlnet-aux==0.0.7
imageio
imageio[ffmpeg]
safetensors
einops
sentencepiece
protobuf
modelscope
ftfy
cupy-cuda12x
absl-py
peft
lightning
pandas
deepspeed
wandb
av
huggingface_hub
torch>=2.0.0
torchvision
transformers==4.46.2
controlnet-aux==0.0.7
imageio
imageio[ffmpeg]
safetensors
einops
sentencepiece
protobuf
modelscope
ftfy
cupy-cuda12x
absl-py
peft
lightning
pandas
deepspeed
wandb
av
huggingface_hub