Unverified Commit 8d36d5ad authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

Update clip_guided_stable_diffusion.py

parent dc2a1c1d
......@@ -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)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment