Unverified Commit 7bc2fff1 authored by Eugene Lyapustin's avatar Eugene Lyapustin Committed by GitHub
Browse files

Fix StableUnCLIPImg2ImgPipeline handling of explicitly passed image embeddings (#2845)

parent 4c26cb9c
...@@ -388,7 +388,7 @@ class StableUnCLIPImg2ImgPipeline(DiffusionPipeline): ...@@ -388,7 +388,7 @@ class StableUnCLIPImg2ImgPipeline(DiffusionPipeline):
# what the expected dimensions of inputs should be and how we handle the encoding. # what the expected dimensions of inputs should be and how we handle the encoding.
repeat_by = num_images_per_prompt repeat_by = num_images_per_prompt
if not image_embeds: if image_embeds is None:
if not isinstance(image, torch.Tensor): if not isinstance(image, torch.Tensor):
image = self.feature_extractor(images=image, return_tensors="pt").pixel_values image = self.feature_extractor(images=image, return_tensors="pt").pixel_values
......
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