Unverified Commit 8ceec90d authored by victolee0's avatar victolee0 Committed by GitHub
Browse files

fix check_inputs func in LuminaText2ImgPipeline (#10651)

parent 158c5c4d
...@@ -396,8 +396,10 @@ class LuminaText2ImgPipeline(DiffusionPipeline): ...@@ -396,8 +396,10 @@ class LuminaText2ImgPipeline(DiffusionPipeline):
prompt_attention_mask=None, prompt_attention_mask=None,
negative_prompt_attention_mask=None, negative_prompt_attention_mask=None,
): ):
if height % 8 != 0 or width % 8 != 0: if height % (self.vae_scale_factor * 2) != 0 or width % (self.vae_scale_factor * 2) != 0:
raise ValueError(f"`height` and `width` have to be divisible by 8 but are {height} and {width}.") raise ValueError(
f"`height` and `width` have to be divisible by {self.vae_scale_factor * 2} but are {height} and {width}."
)
if prompt is not None and prompt_embeds is not None: if prompt is not None and prompt_embeds is not None:
raise ValueError( raise ValueError(
......
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