JiayiGuo821 commited on
Commit
a534a2d
1 Parent(s): 585cedb
Files changed (1) hide show
  1. requirements.txt +3 -4
requirements.txt CHANGED
@@ -12,8 +12,7 @@ pypatchify==0.1.4
12
  safetensors==0.3.1
13
  tqdm==4.65.0
14
  transformers==4.30.1
15
- torch==2.0.0
16
- torchvision==0.15.0
17
- torchaudio==2.0.0
18
  wandb==0.16.3
19
- xformers==0.0.17
 
12
  safetensors==0.3.1
13
  tqdm==4.65.0
14
  transformers==4.30.1
15
+ torch
16
+ torchvision
17
+ torchaudio
18
  wandb==0.16.3