├── .gitignore ├── LICENSE ├── README.md ├── assets ├── 1_repeat-0.mp4 ├── 2_repeat-0.mp4 ├── 3_repeat-0.mp4 ├── 4_repeat-0.mp4 └── demo.mp4 ├── configs ├── accelerate.yaml └── test │ └── 4_step_t2v.yaml ├── datasets ├── __init__.py ├── dataset_utils.py └── test_image.py ├── model_dit ├── __init__.py ├── lightning │ └── magic_141_t2v.py ├── models │ └── magic_141_video │ │ ├── __init__.py │ │ ├── config.py │ │ ├── constants.py │ │ ├── diffusion │ │ ├── __init__.py │ │ └── schedulers │ │ │ ├── __init__.py │ │ │ └── scheduling_flow_match_discrete.py │ │ ├── modules │ │ ├── __init__.py │ │ ├── activation_layers.py │ │ ├── attenion.py │ │ ├── attenion_flashatt3.py │ │ ├── embed_layers.py │ │ ├── mlp_layers.py │ │ ├── models.py │ │ ├── modulate_layers.py │ │ ├── norm_layers.py │ │ ├── posemb_layers.py │ │ └── token_refiner.py │ │ ├── text_encoder │ │ ├── __init__.py │ │ └── text_encoder_vlm.py │ │ ├── utils │ │ ├── __init__.py │ │ ├── data_utils.py │ │ ├── file_utils.py │ │ ├── helpers.py │ │ └── preprocess_text_encoder_tokenizer_utils.py │ │ └── vae │ │ ├── __init__.py │ │ ├── autoencoder_kl_causal_3d.py │ │ ├── unet_causal_3d_blocks.py │ │ └── vae.py ├── modules │ ├── modeling_utils.py │ ├── normalization.py │ ├── patch_utils.py │ └── positional_embd.py ├── utils │ ├── __init__.py │ ├── callbacks.py │ ├── image.py │ ├── loss.py │ ├── quant.py │ └── util.py └── vae │ ├── common │ ├── .DS_Store │ ├── __init__.py │ ├── accumulator.py │ ├── config.py │ ├── decorators.py │ ├── distributed │ │ ├── .DS_Store │ │ ├── __init__.py │ │ ├── advanced.py │ │ ├── basic.py │ │ └── ops.py │ ├── ema.py │ ├── entrypoint.py │ ├── evaluation │ │ ├── clip_score.yaml │ │ ├── fid.yaml │ │ ├── fvd.yaml │ │ ├── lpips.yaml │ │ ├── metrics.py │ │ └── psnr.yaml │ ├── evaluator.py │ ├── fs.py │ ├── logger.py │ ├── loss │ │ └── adversarial.py │ ├── mfu │ │ ├── .DS_Store │ │ ├── __init__.py │ │ ├── accumulator.py │ │ ├── basic_hooks.py │ │ ├── modules.py │ │ ├── state.py │ │ └── utils.py │ ├── partition.py │ ├── persistence │ │ ├── README.md │ │ ├── __init__.py │ │ ├── dataclass.py │ │ ├── distributed.py │ │ ├── loader.py │ │ ├── manager.py │ │ ├── mixin.py │ │ └── utils.py │ ├── platform.py │ ├── precision.py │ ├── profiler.py │ ├── schedules.py │ ├── seed.py │ ├── trainer.py │ └── writers │ │ ├── __init__.py │ │ ├── base.py │ │ ├── collection.py │ │ ├── dummy.py │ │ ├── mixin.py │ │ ├── tensorboard.py │ │ ├── utils.py │ │ └── wandb.py │ └── models │ ├── .DS_Store │ ├── video_ae │ ├── act.py │ ├── ae.py │ ├── blocks.py │ ├── configs │ │ ├── image_ae.yaml │ │ └── video_ae.yaml │ ├── conv.py │ ├── discriminators.py │ ├── distribution.py │ ├── norm.py │ ├── pad.py │ ├── resample.py │ ├── sequential.py │ └── types.py │ └── video_vae │ └── modules │ ├── discriminators.py │ ├── inflated_layers.py │ ├── inflated_lib.py │ └── video_vae.py ├── requirements.txt ├── scripts ├── run_flashatt3.sh └── run_quant.sh └── test_t2v.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/README.md -------------------------------------------------------------------------------- /assets/1_repeat-0.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/assets/1_repeat-0.mp4 -------------------------------------------------------------------------------- /assets/2_repeat-0.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/assets/2_repeat-0.mp4 -------------------------------------------------------------------------------- /assets/3_repeat-0.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/assets/3_repeat-0.mp4 -------------------------------------------------------------------------------- /assets/4_repeat-0.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/assets/4_repeat-0.mp4 -------------------------------------------------------------------------------- /assets/demo.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/assets/demo.mp4 -------------------------------------------------------------------------------- /configs/accelerate.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/configs/accelerate.yaml -------------------------------------------------------------------------------- /configs/test/4_step_t2v.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/configs/test/4_step_t2v.yaml -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/dataset_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/datasets/dataset_utils.py -------------------------------------------------------------------------------- /datasets/test_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/datasets/test_image.py -------------------------------------------------------------------------------- /model_dit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model_dit/lightning/magic_141_t2v.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/lightning/magic_141_t2v.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/config.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/constants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/constants.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/diffusion/__init__.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/diffusion/schedulers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/diffusion/schedulers/__init__.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/diffusion/schedulers/scheduling_flow_match_discrete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/diffusion/schedulers/scheduling_flow_match_discrete.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/modules/__init__.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/modules/activation_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/modules/activation_layers.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/modules/attenion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/modules/attenion.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/modules/attenion_flashatt3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/modules/attenion_flashatt3.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/modules/embed_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/modules/embed_layers.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/modules/mlp_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/modules/mlp_layers.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/modules/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/modules/models.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/modules/modulate_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/modules/modulate_layers.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/modules/norm_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/modules/norm_layers.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/modules/posemb_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/modules/posemb_layers.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/modules/token_refiner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/modules/token_refiner.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/text_encoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/text_encoder/__init__.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/text_encoder/text_encoder_vlm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/text_encoder/text_encoder_vlm.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/utils/data_utils.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/utils/file_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/utils/file_utils.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/utils/helpers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/utils/helpers.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/utils/preprocess_text_encoder_tokenizer_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/utils/preprocess_text_encoder_tokenizer_utils.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/vae/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/vae/__init__.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/vae/autoencoder_kl_causal_3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/vae/autoencoder_kl_causal_3d.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/vae/unet_causal_3d_blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/vae/unet_causal_3d_blocks.py -------------------------------------------------------------------------------- /model_dit/models/magic_141_video/vae/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/models/magic_141_video/vae/vae.py -------------------------------------------------------------------------------- /model_dit/modules/modeling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/modules/modeling_utils.py -------------------------------------------------------------------------------- /model_dit/modules/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/modules/normalization.py -------------------------------------------------------------------------------- /model_dit/modules/patch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/modules/patch_utils.py -------------------------------------------------------------------------------- /model_dit/modules/positional_embd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/modules/positional_embd.py -------------------------------------------------------------------------------- /model_dit/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model_dit/utils/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/utils/callbacks.py -------------------------------------------------------------------------------- /model_dit/utils/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/utils/image.py -------------------------------------------------------------------------------- /model_dit/utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/utils/loss.py -------------------------------------------------------------------------------- /model_dit/utils/quant.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/utils/quant.py -------------------------------------------------------------------------------- /model_dit/utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/utils/util.py -------------------------------------------------------------------------------- /model_dit/vae/common/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/.DS_Store -------------------------------------------------------------------------------- /model_dit/vae/common/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model_dit/vae/common/accumulator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/accumulator.py -------------------------------------------------------------------------------- /model_dit/vae/common/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/config.py -------------------------------------------------------------------------------- /model_dit/vae/common/decorators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/decorators.py -------------------------------------------------------------------------------- /model_dit/vae/common/distributed/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/distributed/.DS_Store -------------------------------------------------------------------------------- /model_dit/vae/common/distributed/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/distributed/__init__.py -------------------------------------------------------------------------------- /model_dit/vae/common/distributed/advanced.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/distributed/advanced.py -------------------------------------------------------------------------------- /model_dit/vae/common/distributed/basic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/distributed/basic.py -------------------------------------------------------------------------------- /model_dit/vae/common/distributed/ops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/distributed/ops.py -------------------------------------------------------------------------------- /model_dit/vae/common/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/ema.py -------------------------------------------------------------------------------- /model_dit/vae/common/entrypoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/entrypoint.py -------------------------------------------------------------------------------- /model_dit/vae/common/evaluation/clip_score.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/evaluation/clip_score.yaml -------------------------------------------------------------------------------- /model_dit/vae/common/evaluation/fid.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/evaluation/fid.yaml -------------------------------------------------------------------------------- /model_dit/vae/common/evaluation/fvd.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/evaluation/fvd.yaml -------------------------------------------------------------------------------- /model_dit/vae/common/evaluation/lpips.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/evaluation/lpips.yaml -------------------------------------------------------------------------------- /model_dit/vae/common/evaluation/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/evaluation/metrics.py -------------------------------------------------------------------------------- /model_dit/vae/common/evaluation/psnr.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/evaluation/psnr.yaml -------------------------------------------------------------------------------- /model_dit/vae/common/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/evaluator.py -------------------------------------------------------------------------------- /model_dit/vae/common/fs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/fs.py -------------------------------------------------------------------------------- /model_dit/vae/common/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/logger.py -------------------------------------------------------------------------------- /model_dit/vae/common/loss/adversarial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/loss/adversarial.py -------------------------------------------------------------------------------- /model_dit/vae/common/mfu/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/mfu/.DS_Store -------------------------------------------------------------------------------- /model_dit/vae/common/mfu/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/mfu/__init__.py -------------------------------------------------------------------------------- /model_dit/vae/common/mfu/accumulator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/mfu/accumulator.py -------------------------------------------------------------------------------- /model_dit/vae/common/mfu/basic_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/mfu/basic_hooks.py -------------------------------------------------------------------------------- /model_dit/vae/common/mfu/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/mfu/modules.py -------------------------------------------------------------------------------- /model_dit/vae/common/mfu/state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/mfu/state.py -------------------------------------------------------------------------------- /model_dit/vae/common/mfu/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/mfu/utils.py -------------------------------------------------------------------------------- /model_dit/vae/common/partition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/partition.py -------------------------------------------------------------------------------- /model_dit/vae/common/persistence/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/persistence/README.md -------------------------------------------------------------------------------- /model_dit/vae/common/persistence/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/persistence/__init__.py -------------------------------------------------------------------------------- /model_dit/vae/common/persistence/dataclass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/persistence/dataclass.py -------------------------------------------------------------------------------- /model_dit/vae/common/persistence/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/persistence/distributed.py -------------------------------------------------------------------------------- /model_dit/vae/common/persistence/loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/persistence/loader.py -------------------------------------------------------------------------------- /model_dit/vae/common/persistence/manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/persistence/manager.py -------------------------------------------------------------------------------- /model_dit/vae/common/persistence/mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/persistence/mixin.py -------------------------------------------------------------------------------- /model_dit/vae/common/persistence/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/persistence/utils.py -------------------------------------------------------------------------------- /model_dit/vae/common/platform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/platform.py -------------------------------------------------------------------------------- /model_dit/vae/common/precision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/precision.py -------------------------------------------------------------------------------- /model_dit/vae/common/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/profiler.py -------------------------------------------------------------------------------- /model_dit/vae/common/schedules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/schedules.py -------------------------------------------------------------------------------- /model_dit/vae/common/seed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/seed.py -------------------------------------------------------------------------------- /model_dit/vae/common/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/trainer.py -------------------------------------------------------------------------------- /model_dit/vae/common/writers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/writers/__init__.py -------------------------------------------------------------------------------- /model_dit/vae/common/writers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/writers/base.py -------------------------------------------------------------------------------- /model_dit/vae/common/writers/collection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/writers/collection.py -------------------------------------------------------------------------------- /model_dit/vae/common/writers/dummy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/writers/dummy.py -------------------------------------------------------------------------------- /model_dit/vae/common/writers/mixin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/writers/mixin.py -------------------------------------------------------------------------------- /model_dit/vae/common/writers/tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/writers/tensorboard.py -------------------------------------------------------------------------------- /model_dit/vae/common/writers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/writers/utils.py -------------------------------------------------------------------------------- /model_dit/vae/common/writers/wandb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/common/writers/wandb.py -------------------------------------------------------------------------------- /model_dit/vae/models/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/.DS_Store -------------------------------------------------------------------------------- /model_dit/vae/models/video_ae/act.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/video_ae/act.py -------------------------------------------------------------------------------- /model_dit/vae/models/video_ae/ae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/video_ae/ae.py -------------------------------------------------------------------------------- /model_dit/vae/models/video_ae/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/video_ae/blocks.py -------------------------------------------------------------------------------- /model_dit/vae/models/video_ae/configs/image_ae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/video_ae/configs/image_ae.yaml -------------------------------------------------------------------------------- /model_dit/vae/models/video_ae/configs/video_ae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/video_ae/configs/video_ae.yaml -------------------------------------------------------------------------------- /model_dit/vae/models/video_ae/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/video_ae/conv.py -------------------------------------------------------------------------------- /model_dit/vae/models/video_ae/discriminators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/video_ae/discriminators.py -------------------------------------------------------------------------------- /model_dit/vae/models/video_ae/distribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/video_ae/distribution.py -------------------------------------------------------------------------------- /model_dit/vae/models/video_ae/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/video_ae/norm.py -------------------------------------------------------------------------------- /model_dit/vae/models/video_ae/pad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/video_ae/pad.py -------------------------------------------------------------------------------- /model_dit/vae/models/video_ae/resample.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/video_ae/resample.py -------------------------------------------------------------------------------- /model_dit/vae/models/video_ae/sequential.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/video_ae/sequential.py -------------------------------------------------------------------------------- /model_dit/vae/models/video_ae/types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/video_ae/types.py -------------------------------------------------------------------------------- /model_dit/vae/models/video_vae/modules/discriminators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/video_vae/modules/discriminators.py -------------------------------------------------------------------------------- /model_dit/vae/models/video_vae/modules/inflated_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/video_vae/modules/inflated_layers.py -------------------------------------------------------------------------------- /model_dit/vae/models/video_vae/modules/inflated_lib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/video_vae/modules/inflated_lib.py -------------------------------------------------------------------------------- /model_dit/vae/models/video_vae/modules/video_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/model_dit/vae/models/video_vae/modules/video_vae.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/run_flashatt3.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/scripts/run_flashatt3.sh -------------------------------------------------------------------------------- /scripts/run_quant.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/scripts/run_quant.sh -------------------------------------------------------------------------------- /test_t2v.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Open-Magic-Video/Magic-1-For-1/HEAD/test_t2v.py --------------------------------------------------------------------------------