diff --git a/examples/community/clip_guided_stable_diffusion.py b/examples/community/clip_guided_stable_diffusion.py index db08a124d9..b24f987ded 100644 --- a/examples/community/clip_guided_stable_diffusion.py +++ b/examples/community/clip_guided_stable_diffusion.py @@ -148,7 +148,7 @@ class CLIPGuidedStableDiffusion(DiffusionPipeline): image = transforms.Resize(self.feature_extractor.size)(image) image = self.normalize(image) - image_embeddings_clip = self.clip_model.get_image_features(image).float() + 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) if use_cutouts: