Spaces:
Running
on
Zero
Running
on
Zero
update dependencies
Browse files- app.py +3 -1
- requirements.txt +7 -1
app.py
CHANGED
|
@@ -1,4 +1,6 @@
|
|
| 1 |
import torch
|
|
|
|
|
|
|
| 2 |
import spaces
|
| 3 |
import numpy as np
|
| 4 |
from diffusers import (
|
|
@@ -33,7 +35,7 @@ apply_normal = NormalDetector(
|
|
| 33 |
apply_segm = SegmDetector()
|
| 34 |
|
| 35 |
|
| 36 |
-
@spaces.GPU(duration=
|
| 37 |
def generate(
|
| 38 |
input_image,
|
| 39 |
prompt,
|
|
|
|
| 1 |
import torch
|
| 2 |
+
torch.jit.script = lambda f: f
|
| 3 |
+
|
| 4 |
import spaces
|
| 5 |
import numpy as np
|
| 6 |
from diffusers import (
|
|
|
|
| 35 |
apply_segm = SegmDetector()
|
| 36 |
|
| 37 |
|
| 38 |
+
@spaces.GPU(duration=20)
|
| 39 |
def generate(
|
| 40 |
input_image,
|
| 41 |
prompt,
|
requirements.txt
CHANGED
|
@@ -1,7 +1,13 @@
|
|
|
|
|
|
|
|
| 1 |
torchvision
|
|
|
|
|
|
|
| 2 |
transformers
|
| 3 |
diffusers
|
| 4 |
opencv-python-headless
|
|
|
|
| 5 |
pillow
|
|
|
|
| 6 |
geffnet
|
| 7 |
-
glob2
|
|
|
|
| 1 |
+
--extra-index-url https://download.pytorch.org/whl/cu121
|
| 2 |
+
torch
|
| 3 |
torchvision
|
| 4 |
+
torchaudio
|
| 5 |
+
|
| 6 |
transformers
|
| 7 |
diffusers
|
| 8 |
opencv-python-headless
|
| 9 |
+
safetensors
|
| 10 |
pillow
|
| 11 |
+
accelerate
|
| 12 |
geffnet
|
| 13 |
+
glob2
|