Spaces:
Running
Running
Commit
·
fb1d381
1
Parent(s):
6912862
update
Browse files- 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",
|