Dan Biagini commited on
Commit
abbab1b
·
1 Parent(s): 624b55e

changing torch to CUDA

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -76,9 +76,9 @@ tenacity==8.5.0
76
  thinc==8.2.5
77
  threadpoolctl==3.5.0
78
  toml==0.10.2
79
- torch==2.4.1+cpu
80
- torchaudio==2.4.1+cpu
81
- torchvision==0.19.1+cpu
82
  tornado==6.4.1
83
  tqdm==4.66.5
84
  typer==0.12.5
 
76
  thinc==8.2.5
77
  threadpoolctl==3.5.0
78
  toml==0.10.2
79
+ torch==2.4.1
80
+ torchaudio==2.4.1
81
+ torchvision==0.19.1
82
  tornado==6.4.1
83
  tqdm==4.66.5
84
  typer==0.12.5