Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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."
|