Unverified Commit a6b843a7 authored by SahilCarterr's avatar SahilCarterr Committed by GitHub
Browse files

[FIX] check_inputs function in lumina2 (#10784)

parent 27b90235
......@@ -379,7 +379,9 @@ class Lumina2Text2ImgPipeline(DiffusionPipeline):
max_sequence_length=None,
):
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 callback_on_step_end_tensor_inputs is not None and not all(
k in self._callback_tensor_inputs for k in callback_on_step_end_tensor_inputs
......
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