Update clip_guided_stable_diffusion.py

2022-09-19 18:03:00 +02:00 · 2022-09-19 18:03:00 +02:00 · 8d36d5adb1
parent dc2a1c1d07
commit 8d36d5adb1
1 changed files with 1 additions and 1 deletions
--- a/examples/community/clip_guided_stable_diffusion.py
+++ b/examples/community/clip_guided_stable_diffusion.py
@ -213,7 +213,7 @@ class CLIPGuidedStableDiffusion(DiffusionPipeline):
                    max_length=self.tokenizer.model_max_length,
                    truncation=True,
                    return_tensors="pt",
-                ).to(self.device)
+                ).input_ids.to(self.device)
            else:
                clip_text_input = text_input.input_ids.to(self.device)
            text_embeddings_clip = self.clip_model.get_text_features(clip_text_input)