erwold commited on
Commit
5e2c351
1 Parent(s): 1b2dbb1

Initial Commit

Browse files
Files changed (2) hide show
  1. README.md +2 -2
  2. requirements.txt +2 -5
README.md CHANGED
@@ -6,6 +6,6 @@ colorTo: gray
6
  sdk: gradio
7
  sdk_version: 5.6.0
8
  app_file: app.py
9
- suggested_hardware: a100-large
10
- python_version: 3.10
11
  ---
 
6
  sdk: gradio
7
  sdk_version: 5.6.0
8
  app_file: app.py
9
+ python_version: "3.10"
10
+ pinned: false
11
  ---
requirements.txt CHANGED
@@ -1,16 +1,13 @@
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
8
-
9
  # Image processing
10
  Pillow>=10.0.0
11
  opencv-python-headless>=4.8.0
12
  numpy>=1.24.0
13
-
14
  # Utilities
15
  protobuf==4.23.4
16
  sentencepiece==0.2.0
 
1
  # Core dependencies
2
+ torch==2.4.0
3
+ torchvision==0.19.0
 
4
  transformers==4.45.0
5
  diffusers==0.30.0
6
  accelerate==0.33.0
 
7
  # Image processing
8
  Pillow>=10.0.0
9
  opencv-python-headless>=4.8.0
10
  numpy>=1.24.0
 
11
  # Utilities
12
  protobuf==4.23.4
13
  sentencepiece==0.2.0