ford442 commited on
Commit
95e446e
·
verified ·
1 Parent(s): 8680037

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -200,10 +200,7 @@ def infer(
200
  else:
201
  enhanced_prompt = prompt
202
  enhanced_prompt_2 = prompt
203
- #model.to('cpu')
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,