rahul7star commited on
Commit
1a4ff0e
·
verified ·
1 Parent(s): 5a0901c

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +21 -2
requirements.txt CHANGED
@@ -44,5 +44,24 @@ av
44
 
45
  huggingface_hub
46
 
47
- # Local module
48
- -e ./PusaV1
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
44
 
45
  huggingface_hub
46
 
47
+ torch>=2.0.0
48
+ torchvision
49
+ transformers==4.46.2
50
+ controlnet-aux==0.0.7
51
+ imageio
52
+ imageio[ffmpeg]
53
+ safetensors
54
+ einops
55
+ sentencepiece
56
+ protobuf
57
+ modelscope
58
+ ftfy
59
+ cupy-cuda12x
60
+ absl-py
61
+ peft
62
+ lightning
63
+ pandas
64
+ deepspeed
65
+ wandb
66
+ av
67
+ huggingface_hub