JiayiGuo821 commited on
Commit
585cedb
1 Parent(s): de12cda
Files changed (1) hide show
  1. requirements.txt +1 -0
requirements.txt CHANGED
@@ -12,6 +12,7 @@ pypatchify==0.1.4
12
  safetensors==0.3.1
13
  tqdm==4.65.0
14
  transformers==4.30.1
 
15
  torchvision==0.15.0
16
  torchaudio==2.0.0
17
  wandb==0.16.3
 
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