Skip to content

Commit 72eae64

Browse files
Fix dtype safety checker inpaint legacy (#1137)
* [Stable Diffusion Inpaint Legacy] Fiix some things * uP
1 parent de75362 commit 72eae64

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -414,7 +414,9 @@ def __call__(
414414
safety_checker_input = self.feature_extractor(self.numpy_to_pil(image), return_tensors="pt").to(
415415
self.device
416416
)
417-
image, has_nsfw_concept = self.safety_checker(images=image, clip_input=safety_checker_input.pixel_values)
417+
image, has_nsfw_concept = self.safety_checker(
418+
images=image, clip_input=safety_checker_input.pixel_values.to(text_embeddings.dtype)
419+
)
418420
else:
419421
has_nsfw_concept = None
420422

0 commit comments

Comments
 (0)