Update app.py
Browse files
app.py
CHANGED
|
@@ -200,10 +200,7 @@ def infer(
|
|
| 200 |
else:
|
| 201 |
enhanced_prompt = prompt
|
| 202 |
enhanced_prompt_2 = prompt
|
| 203 |
-
|
| 204 |
-
global model
|
| 205 |
-
del model
|
| 206 |
-
gc.collect()
|
| 207 |
if latent_file: # Check if a latent file is provided
|
| 208 |
# initial_latents = pipe.prepare_latents(
|
| 209 |
# batch_size=1,
|
|
|
|
| 200 |
else:
|
| 201 |
enhanced_prompt = prompt
|
| 202 |
enhanced_prompt_2 = prompt
|
| 203 |
+
model.to('cpu')
|
|
|
|
|
|
|
|
|
|
| 204 |
if latent_file: # Check if a latent file is provided
|
| 205 |
# initial_latents = pipe.prepare_latents(
|
| 206 |
# batch_size=1,
|