[feature] Add Matrix Game 2.0 training (#1017)

The CI tests are quite unstable, but since multiple CI tests indicates that each individual tests are passed, I think we can merge this.
This commit is contained in:
Kaiqin Kong
2026-01-26 19:21:43 -08:00
committed by GitHub
parent e8c854bcf1
commit 0c6862ca55
25 changed files with 5891 additions and 74 deletions
+1 -2
View File
@@ -1,6 +1,5 @@
from fastvideo import VideoGenerator
from fastvideo.configs.pipelines.wan import MatrixGameI2V480PConfig
from fastvideo.models.dits.matrix_game.utils import create_action_presets
from fastvideo.models.dits.matrixgame.utils import create_action_presets
import torch