qyoo commited on
Commit
1af3dca
·
1 Parent(s): 429e2fd
app.py CHANGED
@@ -67,6 +67,7 @@ pipe = StableDiffusionCustomPipeline.from_pretrained(
67
  )
68
  pipeline = ConceptrolIPAdapterPlus(pipe, "", adapter_name, device, num_tokens=16)
69
 
 
70
  def change_model_fn(model_name: str) -> None:
71
  global device, pipeline
72
 
@@ -186,7 +187,7 @@ def change_generate_button_fn(enable: int) -> gr.Button:
186
  def dynamic_gallery_fn():
187
  return gr.Image(label="Result", show_label=False)
188
 
189
- @spaces.GPU(duration=110)
190
  @torch.no_grad()
191
  def generate(
192
  prompt="a statue is reading the book in the cafe",
 
67
  )
68
  pipeline = ConceptrolIPAdapterPlus(pipe, "", adapter_name, device, num_tokens=16)
69
 
70
+ @spaces.GPU
71
  def change_model_fn(model_name: str) -> None:
72
  global device, pipeline
73
 
 
187
  def dynamic_gallery_fn():
188
  return gr.Image(label="Result", show_label=False)
189
 
190
+ @spaces.GPU
191
  @torch.no_grad()
192
  def generate(
193
  prompt="a statue is reading the book in the cafe",
omini_control/__pycache__/block.cpython-310.pyc CHANGED
Binary files a/omini_control/__pycache__/block.cpython-310.pyc and b/omini_control/__pycache__/block.cpython-310.pyc differ
 
omini_control/__pycache__/condition.cpython-310.pyc CHANGED
Binary files a/omini_control/__pycache__/condition.cpython-310.pyc and b/omini_control/__pycache__/condition.cpython-310.pyc differ
 
omini_control/__pycache__/transformer.cpython-310.pyc CHANGED
Binary files a/omini_control/__pycache__/transformer.cpython-310.pyc and b/omini_control/__pycache__/transformer.cpython-310.pyc differ