ltx: vae: Fix missing init variable (#13074)
Forgot to push this ammendment. Previous test results apply to this.pull/13060/head^2
parent
82b868a45a
commit
f49856af57
|
|
@ -602,6 +602,7 @@ class Decoder(nn.Module):
|
|||
)
|
||||
|
||||
timestep_shift_scale = None
|
||||
scaled_timestep = None
|
||||
if self.timestep_conditioning:
|
||||
assert (
|
||||
timestep is not None
|
||||
|
|
|
|||
Loading…
Reference in New Issue