Spaces:
Running
on
Zero
Running
on
Zero
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,
|