ltx: vae: Fix missing init variable

Forgot to push this ammendment. Previous test results apply to this.
This commit is contained in:
Rattus 2026-03-20 12:09:24 +10:00
parent 82b868a45a
commit 2d1db34092

View File

@ -602,6 +602,7 @@ class Decoder(nn.Module):
)
timestep_shift_scale = None
scaled_timestep = None
if self.timestep_conditioning:
assert (
timestep is not None