Spaces:
Runtime error
Runtime error
wondervictor
commited on
Commit
·
995b367
1
Parent(s):
66ffc69
add requirements
Browse files- app_canny.py +2 -2
- app_depth.py +1 -1
- model.py +2 -2
app_canny.py
CHANGED
|
@@ -10,12 +10,12 @@ def randomize_seed_fn(seed: int, randomize_seed: bool) -> int:
|
|
| 10 |
|
| 11 |
examples = [
|
| 12 |
[
|
| 13 |
-
"condition/example/t2i/multigen/doll.
|
| 14 |
"A stuffed animal wearing a mask and a leash, sitting on a blanket",
|
| 15 |
"(512, 512)"
|
| 16 |
],
|
| 17 |
[
|
| 18 |
-
"condition/example/t2i/multigen/girl.
|
| 19 |
"An anime style girl with blue hair", "(512, 512)"
|
| 20 |
],
|
| 21 |
[
|
|
|
|
| 10 |
|
| 11 |
examples = [
|
| 12 |
[
|
| 13 |
+
"condition/example/t2i/multigen/doll.jpg",
|
| 14 |
"A stuffed animal wearing a mask and a leash, sitting on a blanket",
|
| 15 |
"(512, 512)"
|
| 16 |
],
|
| 17 |
[
|
| 18 |
+
"condition/example/t2i/multigen/girl.jpg",
|
| 19 |
"An anime style girl with blue hair", "(512, 512)"
|
| 20 |
],
|
| 21 |
[
|
app_depth.py
CHANGED
|
@@ -13,7 +13,7 @@ examples = [[
|
|
| 13 |
"The red sofa in the living room has several pillows on it", "(512, 512)"
|
| 14 |
],
|
| 15 |
[
|
| 16 |
-
"condition/example/t2i/multigen/house.
|
| 17 |
"A brick house with a chimney under a starry sky.",
|
| 18 |
"(512, 512)"
|
| 19 |
],
|
|
|
|
| 13 |
"The red sofa in the living room has several pillows on it", "(512, 512)"
|
| 14 |
],
|
| 15 |
[
|
| 16 |
+
"condition/example/t2i/multigen/house.jpg",
|
| 17 |
"A brick house with a chimney under a starry sky.",
|
| 18 |
"(512, 512)"
|
| 19 |
],
|
model.py
CHANGED
|
@@ -92,7 +92,7 @@ class Model:
|
|
| 92 |
return t5_model
|
| 93 |
|
| 94 |
@torch.no_grad()
|
| 95 |
-
@spaces.GPU(enable_queue=
|
| 96 |
def process_canny(
|
| 97 |
self,
|
| 98 |
image: np.ndarray,
|
|
@@ -168,7 +168,7 @@ class Model:
|
|
| 168 |
return samples
|
| 169 |
|
| 170 |
@torch.no_grad()
|
| 171 |
-
@spaces.GPU(enable_queue=
|
| 172 |
def process_depth(
|
| 173 |
self,
|
| 174 |
image: np.ndarray,
|
|
|
|
| 92 |
return t5_model
|
| 93 |
|
| 94 |
@torch.no_grad()
|
| 95 |
+
@spaces.GPU(enable_queue=False)
|
| 96 |
def process_canny(
|
| 97 |
self,
|
| 98 |
image: np.ndarray,
|
|
|
|
| 168 |
return samples
|
| 169 |
|
| 170 |
@torch.no_grad()
|
| 171 |
+
@spaces.GPU(enable_queue=False)
|
| 172 |
def process_depth(
|
| 173 |
self,
|
| 174 |
image: np.ndarray,
|