Kaiqin Kongandmergify[bot]
773d44b875
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
15 lines
462 B
Python
15 lines
462 B
Python
from .model import MatrixGame2WanModel, MatrixGame2TransformerBlock
|
|
from .causal_model import CausalMatrixGame2WanModel, CausalMatrixGame2TransformerBlock
|
|
from .action_module import ActionModule
|
|
|
|
__all__ = [
|
|
"MatrixGame2WanModel",
|
|
"MatrixGame2TransformerBlock",
|
|
"CausalMatrixGame2WanModel",
|
|
"CausalMatrixGame2TransformerBlock",
|
|
"ActionModule",
|
|
]
|
|
|
|
# Entry point for model registry
|
|
EntryClass = [MatrixGame2WanModel, CausalMatrixGame2WanModel]
|