Update app.py
Browse files
app.py
CHANGED
|
@@ -70,8 +70,8 @@ def generate_image(enhanced_caption, seed, num_inference_steps, guidance_scale):
|
|
| 70 |
|
| 71 |
print(image)
|
| 72 |
|
| 73 |
-
pipe.to("cpu")
|
| 74 |
-
torch.cuda.empty_cache()
|
| 75 |
|
| 76 |
return image
|
| 77 |
|
|
|
|
| 70 |
|
| 71 |
print(image)
|
| 72 |
|
| 73 |
+
# pipe.to("cpu")
|
| 74 |
+
# torch.cuda.empty_cache()
|
| 75 |
|
| 76 |
return image
|
| 77 |
|