Spaces:
Running
on
Zero
Running
on
Zero
ohayonguy
commited on
Commit
·
05b8f98
1
Parent(s):
a722599
issue with torch version
Browse files- app.py +1 -0
- requirements.txt +2 -2
app.py
CHANGED
|
@@ -2,6 +2,7 @@ import os
|
|
| 2 |
if os.getenv('SPACES_ZERO_GPU') == "true":
|
| 3 |
os.environ['SPACES_ZERO_GPU'] = "1"
|
| 4 |
os.environ['K_DIFFUSION_USE_COMPILE'] = "0"
|
|
|
|
| 5 |
import spaces
|
| 6 |
import cv2
|
| 7 |
import gradio as gr
|
|
|
|
| 2 |
if os.getenv('SPACES_ZERO_GPU') == "true":
|
| 3 |
os.environ['SPACES_ZERO_GPU'] = "1"
|
| 4 |
os.environ['K_DIFFUSION_USE_COMPILE'] = "0"
|
| 5 |
+
import natten
|
| 6 |
import spaces
|
| 7 |
import cv2
|
| 8 |
import gradio as gr
|
requirements.txt
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
-f https://download.pytorch.org/whl/cu118
|
| 2 |
-
torch==2.2.
|
| 3 |
-f https://download.pytorch.org/whl/cu118
|
| 4 |
-
torchvision==0.17.
|
| 5 |
realesrgan==0.2.5
|
| 6 |
numpy
|
| 7 |
facexlib
|
|
|
|
| 1 |
-f https://download.pytorch.org/whl/cu118
|
| 2 |
+
torch==2.2.1
|
| 3 |
-f https://download.pytorch.org/whl/cu118
|
| 4 |
+
torchvision==0.17.1
|
| 5 |
realesrgan==0.2.5
|
| 6 |
numpy
|
| 7 |
facexlib
|