QinOwen commited on
Commit
2846498
1 Parent(s): a9bbd0d

example-lazy

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -60,7 +60,7 @@ def update_dropdown(lora_rank):
60
  else: # 0
61
  return gr.update(value="Base Model")
62
 
63
- @spaces.GPU(duration=120)
64
  def setup_model_progress(lora_model, lora_rank):
65
  global model
66
 
@@ -72,7 +72,7 @@ def setup_model_progress(lora_model, lora_rank):
72
  # Enable buttons after loading and update indicator
73
  yield (gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=True), "Model loaded successfully")
74
 
75
- @spaces.GPU(duration=120)
76
  def generate_example(prompt, lora_model, lora_rank, seed, height, width, unconditional_guidance_scale, ddim_steps, ddim_eta,
77
  frames, savefps):
78
  global model
@@ -267,7 +267,7 @@ with gr.Blocks(css=custom_css) as demo:
267
  outputs=output_video,
268
  fn=generate_example,
269
  run_on_click=False,
270
- cache_examples=True,
271
  )
272
 
273
  demo.launch(share=True)
 
60
  else: # 0
61
  return gr.update(value="Base Model")
62
 
63
+ @spaces.GPU(duration=180)
64
  def setup_model_progress(lora_model, lora_rank):
65
  global model
66
 
 
72
  # Enable buttons after loading and update indicator
73
  yield (gr.update(interactive=True), gr.update(interactive=True), gr.update(interactive=True), "Model loaded successfully")
74
 
75
+ @spaces.GPU(duration=300)
76
  def generate_example(prompt, lora_model, lora_rank, seed, height, width, unconditional_guidance_scale, ddim_steps, ddim_eta,
77
  frames, savefps):
78
  global model
 
267
  outputs=output_video,
268
  fn=generate_example,
269
  run_on_click=False,
270
+ cache_examples="lazy",
271
  )
272
 
273
  demo.launch(share=True)