Update app.py
Browse files
app.py
CHANGED
|
@@ -184,7 +184,7 @@ def generate_image(prompt, negative_prompt, lora_url, num_inference_steps=30, gu
|
|
| 184 |
num_images_per_prompt=num_images,
|
| 185 |
generator=generator
|
| 186 |
)
|
| 187 |
-
|
| 188 |
return result.images,seed
|
| 189 |
|
| 190 |
def clean_lora_cache():
|
|
|
|
| 184 |
num_images_per_prompt=num_images,
|
| 185 |
generator=generator
|
| 186 |
)
|
| 187 |
+
torch.cuda.empty_cache()
|
| 188 |
return result.images,seed
|
| 189 |
|
| 190 |
def clean_lora_cache():
|