From 435433cefdecad09a07d95a73377d528d8caec1a Mon Sep 17 00:00:00 2001 From: Suraj Patil Date: Thu, 6 Oct 2022 18:38:09 +0200 Subject: [PATCH] Update clip_guided_stable_diffusion.py --- examples/community/clip_guided_stable_diffusion.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/community/clip_guided_stable_diffusion.py b/examples/community/clip_guided_stable_diffusion.py index b24f987d..974f4ab2 100644 --- a/examples/community/clip_guided_stable_diffusion.py +++ b/examples/community/clip_guided_stable_diffusion.py @@ -146,7 +146,7 @@ class CLIPGuidedStableDiffusion(DiffusionPipeline): image = self.make_cutouts(image, num_cutouts) else: image = transforms.Resize(self.feature_extractor.size)(image) - image = self.normalize(image) + image = self.normalize(image).to(latents.dtype) image_embeddings_clip = self.clip_model.get_image_features(image) image_embeddings_clip = image_embeddings_clip / image_embeddings_clip.norm(p=2, dim=-1, keepdim=True)