kfahn commited on
Commit
8a59f67
1 Parent(s): 16481ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ def infer(prompts, negative_prompts, image):
66
  #output_images = pipe.numpy_to_pil(np.asarray(output.reshape((num_samples,) + output.shape[-3:])))
67
  del image
68
  del prompt_ids
69
- del neg_prompt_ids
70
  gc.collect()
71
 
72
  output=np.array(output, dtype=np.float32)
 
66
  #output_images = pipe.numpy_to_pil(np.asarray(output.reshape((num_samples,) + output.shape[-3:])))
67
  del image
68
  del prompt_ids
69
+ del negative_prompt_ids
70
  gc.collect()
71
 
72
  output=np.array(output, dtype=np.float32)