Spaces:
Running
Running
cocktailpeanut
commited on
Commit
•
d6db55a
1
Parent(s):
d35773c
update
Browse files- gradio_demo/app.py +1 -0
gradio_demo/app.py
CHANGED
@@ -176,6 +176,7 @@ def instantir_restore(
|
|
176 |
preview_img.append(f"preview_{i}")
|
177 |
|
178 |
gc.collect()
|
|
|
179 |
if torch.cuda.is_available():
|
180 |
torch.cuda.empty_cache()
|
181 |
elif torch.backends.mps.is_available():
|
|
|
176 |
preview_img.append(f"preview_{i}")
|
177 |
|
178 |
gc.collect()
|
179 |
+
print(f"TORCH={torch}")
|
180 |
if torch.cuda.is_available():
|
181 |
torch.cuda.empty_cache()
|
182 |
elif torch.backends.mps.is_available():
|