erwold commited on
Commit
3968707
·
1 Parent(s): fa28a3d

Initial Commit

Browse files
Files changed (1) hide show
  1. requirements.txt +3 -3
requirements.txt CHANGED
@@ -1,6 +1,7 @@
1
  # Core dependencies
2
- torch>=2.4.1
3
- torchvision>=0.15.0
 
4
  transformers==4.45.0
5
  diffusers==0.30.0
6
  accelerate==0.33.0
@@ -13,5 +14,4 @@ numpy>=1.24.0
13
  # Utilities
14
  protobuf==4.23.4
15
  sentencepiece==0.2.0
16
-
17
  gradio==4.36.0
 
1
  # Core dependencies
2
+ --extra-index-url https://download.pytorch.org/whl/cu118
3
+ torch==2.4.1+cu118
4
+ torchvision==0.15.0+cu118
5
  transformers==4.45.0
6
  diffusers==0.30.0
7
  accelerate==0.33.0
 
14
  # Utilities
15
  protobuf==4.23.4
16
  sentencepiece==0.2.0
 
17
  gradio==4.36.0