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

Initial Commit

Browse files
Files changed (1) hide show
  1. requirements.txt +4 -4
requirements.txt CHANGED
@@ -1,7 +1,7 @@
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,4 +14,4 @@ numpy>=1.24.0
14
  # Utilities
15
  protobuf==4.23.4
16
  sentencepiece==0.2.0
17
- gradio==4.36.0
 
1
  # Core dependencies
2
+ --extra-index-url https://download.pytorch.org/whl/cu121
3
+ torch==2.4.1+cu121
4
+ torchvision==0.15.0+cu121
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==5.6.0