Fix dtype safety checker inpaint legacy (#1137)

* [Stable Diffusion Inpaint Legacy] Fiix some things * uP

Fix dtype safety checker inpaint legacy (#1137)
* [Stable Diffusion Inpaint Legacy] Fiix some things * uP
72eae64d · Patrick von Platen · GitHub · de753628 · 72eae64d
Unverified Commit 72eae64d authored Nov 07, 2022 by Patrick von Platen Committed by GitHub Nov 07, 2022
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py ...ble_diffusion/pipeline_stable_diffusion_inpaint_legacy.py +3 -1

No files found.
--- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py
+++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py
@@ -414,7 +414,9 @@ class StableDiffusionInpaintPipelineLegacy(DiffusionPipeline):
            safety_checker_input = self.feature_extractor(self.numpy_to_pil(image), return_tensors="pt").to(
                self.device
            )
-            image, has_nsfw_concept = self.safety_checker(images=image, clip_input=safety_checker_input.pixel_values)
+            image, has_nsfw_concept = self.safety_checker(
+                images=image, clip_input=safety_checker_input.pixel_values.to(text_embeddings.dtype)
+            )
        else:
            has_nsfw_concept = None