fffiloni commited on
Commit
41b0252
·
verified ·
1 Parent(s): 3bffc45

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -205,9 +205,9 @@ tokenizers
205
  tomli
206
  tomlkit
207
  toolz
208
- torch
209
- torchaudio
210
- torchvision
211
  transformers
212
  triton
213
  typer
 
205
  tomli
206
  tomlkit
207
  toolz
208
+ torch==2.0.1
209
+ torchaudio==2.0.2
210
+ torchvision==0.15.2
211
  transformers
212
  triton
213
  typer