15 lines
452 B
Python
15 lines
452 B
Python
from .model import MatrixGameWanModel, MatrixGameTransformerBlock
|
|
from .causal_model import CausalMatrixGameWanModel, CausalMatrixGameTransformerBlock
|
|
from .action_module import ActionModule
|
|
|
|
__all__ = [
|
|
"MatrixGameWanModel",
|
|
"MatrixGameTransformerBlock",
|
|
"CausalMatrixGameWanModel",
|
|
"CausalMatrixGameTransformerBlock",
|
|
"ActionModule",
|
|
]
|
|
|
|
# Entry point for model registry
|
|
EntryClass = [MatrixGameWanModel, CausalMatrixGameWanModel]
|