wondervictor commited on
Commit
995b367
·
1 Parent(s): 66ffc69

add requirements

Browse files
Files changed (3) hide show
  1. app_canny.py +2 -2
  2. app_depth.py +1 -1
  3. 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.png",
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.png",
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.png",
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=True)
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=True)
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,