LPX55 commited on
Commit
28e6562
·
verified ·
1 Parent(s): d971485

Update optimized.py

Browse files
Files changed (1) hide show
  1. 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()