Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -50,7 +50,7 @@ pipe.to(device)
|
|
| 50 |
|
| 51 |
|
| 52 |
|
| 53 |
-
@spaces.GPU()
|
| 54 |
def generate(prompt, negative_prompt, width=1024, height=1024, num_inference_steps=30, lora_id=None, progress=gr.Progress(track_tqdm=True)):
|
| 55 |
if lora_id and lora_id.strip() != "":
|
| 56 |
pipe.unload_lora_weights()
|
|
|
|
| 50 |
|
| 51 |
|
| 52 |
|
| 53 |
+
@spaces.GPU(size='xlarge')
|
| 54 |
def generate(prompt, negative_prompt, width=1024, height=1024, num_inference_steps=30, lora_id=None, progress=gr.Progress(track_tqdm=True)):
|
| 55 |
if lora_id and lora_id.strip() != "":
|
| 56 |
pipe.unload_lora_weights()
|