Skip to content

Commit

Permalink
[FIX] check_inputs function in lumina2
Browse files Browse the repository at this point in the history
  • Loading branch information
SahilCarterr committed Feb 13, 2025
1 parent 051ebc3 commit dbc29bc
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/diffusers/pipelines/lumina2/pipeline_lumina2.py
Original file line number Diff line number Diff line change
Expand Up @@ -379,7 +379,9 @@ def check_inputs(
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
Expand Down

0 comments on commit dbc29bc

Please sign in to comment.