JiayiGuo821 commited on
Commit
9d1eedc
·
1 Parent(s): b6e0092
Files changed (1) hide show
  1. requirements.txt +2 -0
requirements.txt CHANGED
@@ -12,5 +12,7 @@ pypatchify==0.1.4
12
  safetensors==0.3.1
13
  tqdm==4.65.0
14
  transformers==4.30.1
 
 
15
  wandb==0.16.3
16
  xformers==0.0.17
 
12
  safetensors==0.3.1
13
  tqdm==4.65.0
14
  transformers==4.30.1
15
+ torchvision==0.17.0
16
+ torchaudio==2.2.0
17
  wandb==0.16.3
18
  xformers==0.0.17