barakmeiri commited on
Commit
f84c6ef
·
verified ·
1 Parent(s): deb578d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -65,8 +65,8 @@ with gr.Blocks(css="style.css") as demo:
65
  image_editor = image_editor.to('cpu')
66
  if device == 'cuda':
67
  torch.cuda.empty_cache()
68
- print("#### torch.cuda.memory_summary(device=None, abbreviated=False):")
69
- print(torch.cuda.memory_summary(device=None, abbreviated=False))
70
 
71
  if input_image is None or not description_prompt:
72
  return None, "Please set all inputs."
 
65
  image_editor = image_editor.to('cpu')
66
  if device == 'cuda':
67
  torch.cuda.empty_cache()
68
+ # print("#### torch.cuda.memory_summary(device=None, abbreviated=False):")
69
+ # print(torch.cuda.memory_summary(device=None, abbreviated=False))
70
 
71
  if input_image is None or not description_prompt:
72
  return None, "Please set all inputs."