cocktailpeanut commited on
Commit
fb1d381
·
1 Parent(s): 6912862
Files changed (1) hide show
  1. gradio_demo/app.py +2 -0
gradio_demo/app.py CHANGED
@@ -181,8 +181,10 @@ def instantir_restore(
181
  torch.cuda.empty_cache()
182
  elif torch.backends.mps.is_available():
183
  torch.mps.empty_cache()
 
184
  return out[0][0], out[1]
185
 
 
186
  examples = [
187
  "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
188
  "An astronaut riding a green horse",
 
181
  torch.cuda.empty_cache()
182
  elif torch.backends.mps.is_available():
183
  torch.mps.empty_cache()
184
+ gc.collect()
185
  return out[0][0], out[1]
186
 
187
+
188
  examples = [
189
  "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k",
190
  "An astronaut riding a green horse",