Spaces:
Running
on
Zero
Running
on
Zero
Update optimized.py
Browse files- optimized.py +1 -1
optimized.py
CHANGED
@@ -121,7 +121,7 @@ iface = gr.Interface(
|
|
121 |
title="FLUX ControlNet Image Generation",
|
122 |
description="Generate images using the FluxControlNetPipeline. Upload a control image and enter a prompt to create an image.",
|
123 |
)
|
124 |
-
print(f"Memory Usage: {torch.cuda.memory_summary(device=None, abbreviated=False)})
|
125 |
|
126 |
# Launch the app
|
127 |
iface.launch()
|
|
|
121 |
title="FLUX ControlNet Image Generation",
|
122 |
description="Generate images using the FluxControlNetPipeline. Upload a control image and enter a prompt to create an image.",
|
123 |
)
|
124 |
+
print(f"Memory Usage: {torch.cuda.memory_summary(device=None, abbreviated=False)}")
|
125 |
|
126 |
# Launch the app
|
127 |
iface.launch()
|