cocktailpeanut commited on
Commit
d6db55a
1 Parent(s): d35773c
Files changed (1) hide show
  1. 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():