multimodalart HF staff commited on
Commit
bda87aa
1 Parent(s): 161e2f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ pipe_schnell = DiffusionPipeline.from_pretrained(
28
  torch_dtype=torch.bfloat16
29
  )
30
 
31
- @spaces.GPU(duration=75)
32
  def run_parallel_models(prompt, progress=gr.Progress(track_tqdm=True)):
33
  pipe_dev.load_lora_weights(hyper_lora)
34
  image = pipe_dev(prompt, num_inference_steps=8, joint_attention_kwargs={"scale": 0.125}).images[0]
 
28
  torch_dtype=torch.bfloat16
29
  )
30
 
31
+ @spaces.GPU(duration=90)
32
  def run_parallel_models(prompt, progress=gr.Progress(track_tqdm=True)):
33
  pipe_dev.load_lora_weights(hyper_lora)
34
  image = pipe_dev(prompt, num_inference_steps=8, joint_attention_kwargs={"scale": 0.125}).images[0]