SIGMitch commited on
Commit
e5af8e8
·
1 Parent(s): 93e4924
Files changed (1) hide show
  1. latent_consistency_img2img.py +1 -3
latent_consistency_img2img.py CHANGED
@@ -425,9 +425,7 @@ class LatentConsistencyModelImg2ImgPipeline(DiffusionPipeline):
425
  image = self.vae.decode(
426
  denoised / self.vae.config.scaling_factor, return_dict=False
427
  )[0]
428
- image, has_nsfw_concept = self.run_safety_checker(
429
- image, device, prompt_embeds.dtype
430
- )
431
  else:
432
  image = denoised
433
  has_nsfw_concept = None
 
425
  image = self.vae.decode(
426
  denoised / self.vae.config.scaling_factor, return_dict=False
427
  )[0]
428
+ image, has_nsfw_concept = None
 
 
429
  else:
430
  image = denoised
431
  has_nsfw_concept = None