comrender commited on
Commit
43d4e79
·
verified ·
1 Parent(s): c84d7da

Update requirements.txt

Browse files
Files changed (1) hide show
  1. requirements.txt +5 -2
requirements.txt CHANGED
@@ -1,6 +1,8 @@
 
1
  gradio
2
  gradio_imageslider
3
- torch==2.3.0+cu121 # Adjust CUDA version based on your environment
 
4
  diffusers
5
  transformers==4.45.0
6
  huggingface_hub
@@ -10,4 +12,5 @@ safetensors
10
  Pillow
11
  numpy
12
  requests
13
- basicsr
 
 
1
+ --index-url https://download.pytorch.org/whl/cu121
2
  gradio
3
  gradio_imageslider
4
+ torch==2.3.0
5
+ torchvision==0.18.0
6
  diffusers
7
  transformers==4.45.0
8
  huggingface_hub
 
12
  Pillow
13
  numpy
14
  requests
15
+ basicsr
16
+ gfpgan