Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -28,7 +28,7 @@ pipe_schnell = DiffusionPipeline.from_pretrained(
|
|
| 28 |
torch_dtype=torch.bfloat16
|
| 29 |
)
|
| 30 |
|
| 31 |
-
@spaces.GPU(duration=
|
| 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]
|