Spaces:
Paused
Paused
nsfw
Browse files
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 =
|
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
|