style

huggingface · Sep 28, 2024 · f2f2f9c · f2f2f9c
1 parent 2934570
commit f2f2f9c
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/examples/community/matryoshka.py b/examples/community/matryoshka.py
@@ -55,7 +55,6 @@
 from diffusers.pipelines.pipeline_utils import DiffusionPipeline, StableDiffusionMixin
 from diffusers.pipelines.stable_diffusion.pipeline_output import StableDiffusionPipelineOutput
 from diffusers.pipelines.stable_diffusion.safety_checker import StableDiffusionSafetyChecker
-from diffusers.schedulers import KarrasDiffusionSchedulers
 from diffusers.schedulers.scheduling_utils import SchedulerMixin
 from diffusers.utils import (
     USE_PEFT_BACKEND,
@@ -4566,7 +4565,7 @@ def __call__(
                     xm.mark_step()
 
         if not output_type == "latent":
-            image, has_nsfw_concept = self.run_safety_checker(image, device, prompt_embeds.dtype)
+            image, has_nsfw_concept = self.run_safety_checker(latents, device, prompt_embeds.dtype)
         else:
             image = latents
             has_nsfw_concept = None