diffusers
/

matrix-game-2-modular

Model card Files Files and versions

matrix-game-2-modular / block.py

dn6's picture

dn6 HF Staff

Upload folder using huggingface_hub

5178ef1 verified 8 months ago

history blame contribute delete

287 Bytes

	from diffusers.modular_pipelines import SequentialPipelineBlocks
	from .modular_blocks import ACTION2VIDEO_BLOCKS

	class MatrixGameWanBlocks(SequentialPipelineBlocks):
	block_classes = list(ACTION2VIDEO_BLOCKS.copy().values())
	block_names = list(ACTION2VIDEO_BLOCKS.copy().keys())