Add support for MiniMax H3 T2V and I2V training

This commit is contained in:
Jaret Burkett committed 2026-08-03 10:17:39 -06:00
1 parent 73cab2acf5
commit 8502a845b1
28 files changed
+4095 -41

No files matched your search

+13 -1
View File
@@ -274,12 +274,24 @@ class BaseModel:
except:
# if we have a custom vae, it might not have this
divisibility = 8
# flux packs this again,
if self.is_flux:
divisibility = divisibility * 2
return divisibility
def get_frame_count_snapper(self):
"""Optional hook for video models whose VAE accepts frame counts on a
grid other than the default ``temporal_compression * n + 1``.
Return a MODULE-LEVEL function ``(num_frames: int) -> int`` that snaps
an arbitrary frame count DOWN to the nearest count the video VAE can
encode (it must be picklable by reference — file items travel into
dataloader workers, so no lambdas or bound methods). Returning None
keeps the default auto_frame_count behavior.
"""
return None
# these must be implemented in child classes
def load_model(self):
# override this in child classes