from diffusers.modular_pipelines import WanModularPipeline

from modular_blocks import ChronoEditBlocks

repo_id = "nvidia/ChronoEdit-14B-Diffusers"
blocks = ChronoEditBlocks()

blocks.push_to_hub("diffusers-internal-dev/chronoedit-modular")

pipe = WanModularPipeline(blocks, repo_id)
pipe.push_to_hub("diffusers-internal-dev/chronoedit-modular")