Lin Z commited on
Commit
1b4fe4b
1 Parent(s): d6d7648
Files changed (1) hide show
  1. requirements.txt +4 -1
requirements.txt CHANGED
@@ -1,3 +1,6 @@
 
 
 
1
  accelerate==0.32.1
2
  diffusers==0.29.2
3
  einops==0.8.0
@@ -8,4 +11,4 @@ pytorchvideo==0.1.5
8
  timm==1.0.7
9
  tqdm==4.66.4
10
  transformers==4.42.4
11
- wandb==0.17.5
 
1
+ torch==2.1.0
2
+ torchaudio==2.1.0
3
+ torchvision==0.16.0
4
  accelerate==0.32.1
5
  diffusers==0.29.2
6
  einops==0.8.0
 
11
  timm==1.0.7
12
  tqdm==4.66.4
13
  transformers==4.42.4
14
+ gradio==4.39.0