Spaces:
Runtime error
Runtime error
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",
|