Ignore i2v on ltx is training on images
This commit is contained in:
@@ -751,7 +751,7 @@ class LTX2Model(BaseModel):
|
|||||||
video_timestep = timestep.clone()
|
video_timestep = timestep.clone()
|
||||||
|
|
||||||
# i2v from first frame
|
# i2v from first frame
|
||||||
if batch.dataset_config.do_i2v:
|
if batch.dataset_config.do_i2v and batch.dataset_config.num_frames > 1:
|
||||||
# check to see if we had it cached
|
# check to see if we had it cached
|
||||||
if batch.first_frame_latents is not None:
|
if batch.first_frame_latents is not None:
|
||||||
init_latents = batch.first_frame_latents.to(
|
init_latents = batch.first_frame_latents.to(
|
||||||
|
|||||||
Reference in New Issue
Block a user