Update app.py
Browse files
app.py
CHANGED
@@ -86,7 +86,7 @@ def generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale):
|
|
86 |
system_prompt="You are an assistant designed to generate superior images with the superior degree of image-text alignment based on textual prompts or user prompts.",
|
87 |
).images[0]
|
88 |
|
89 |
-
torch.cuda.empty_cache()
|
90 |
|
91 |
return image
|
92 |
|
|
|
86 |
system_prompt="You are an assistant designed to generate superior images with the superior degree of image-text alignment based on textual prompts or user prompts.",
|
87 |
).images[0]
|
88 |
|
89 |
+
# torch.cuda.empty_cache()
|
90 |
|
91 |
return image
|
92 |
|