LICENSE
README.md
pyproject.toml
custom_veomni.egg-info/PKG-INFO
custom_veomni.egg-info/SOURCES.txt
custom_veomni.egg-info/dependency_links.txt
custom_veomni.egg-info/requires.txt
custom_veomni.egg-info/top_level.txt
docs/conf.py
scripts/deepseek_v3/fp8_cast_bf16.py
scripts/deepseek_v3/kernel.py
scripts/moe_ckpt_merge/moe_merge.py
scripts/profile/merge_chrome_trace.py
tasks/omni/infer_omni_model.py
tasks/omni/infer_qwen2_vl.py
tasks/omni/train_flux.py
tasks/omni/train_omni_model.py
tasks/omni/train_qwen2_5_vl.py
tasks/omni/train_qwen2_vl.py
tasks/omni/train_wan.py
veomni/__init__.py
veomni/checkpoint/__init__.py
veomni/checkpoint/checkpointer.py
veomni/checkpoint/format_utils.py
veomni/data/__init__.py
veomni/data/batching_strategy.py
veomni/data/chat_template.py
veomni/data/constants.py
veomni/data/data_collator.py
veomni/data/data_loader.py
veomni/data/data_transform.py
veomni/data/dataset.py
veomni/data/dummy_dataset.py
veomni/data/dynamic_batching.py
veomni/data/diffusion/data_loader.py
veomni/data/diffusion/dataset.py
veomni/data/diffusion/preprocess/origin_dataset.py
veomni/data/multimodal/__init__.py
veomni/data/multimodal/audio_utils.py
veomni/data/multimodal/data_collator.py
veomni/data/multimodal/image_utils.py
veomni/data/multimodal/multimodal_chat_template.py
veomni/data/multimodal/multimodal_transform.py
veomni/data/multimodal/preprocess.py
veomni/data/multimodal/video_utils.py
veomni/distributed/__init__.py
veomni/distributed/checkpoint.py
veomni/distributed/offloading.py
veomni/distributed/parallel_plan.py
veomni/distributed/parallel_state.py
veomni/distributed/torch_parallelize.py
veomni/distributed/utils.py
veomni/distributed/vescale_parallelize.py
veomni/distributed/vescale_plan.py
veomni/distributed/fsdp/__init__.py
veomni/distributed/fsdp/clip_grad_norm.py
veomni/distributed/fsdp/extension.py
veomni/distributed/fsdp/initialize.py
veomni/distributed/fsdp2/__init__.py
veomni/distributed/fsdp2/clip_grad_norm.py
veomni/distributed/moe/__init__.py
veomni/distributed/moe/comm.py
veomni/distributed/moe/moe_layer.py
veomni/distributed/moe/moe_utils.py
veomni/distributed/sequence_parallel/__init__.py
veomni/distributed/sequence_parallel/async_ulysses.py
veomni/distributed/sequence_parallel/async_ulysses_dit.py
veomni/distributed/sequence_parallel/comm.py
veomni/distributed/sequence_parallel/data.py
veomni/distributed/sequence_parallel/loss.py
veomni/distributed/sequence_parallel/ulysses.py
veomni/distributed/sequence_parallel/utils.py
veomni/models/__init__.py
veomni/models/auto.py
veomni/models/loader.py
veomni/models/module_utils.py
veomni/models/registry.py
veomni/models/custom/__init__.py
veomni/models/seed_omni/__init__.py
veomni/models/seed_omni/auto.py
veomni/models/seed_omni/configuration_seed_omni.py
veomni/models/seed_omni/modeling_seed_omni.py
veomni/models/seed_omni/omni_utils.py
veomni/models/seed_omni/processing_seed_omni.py
veomni/models/seed_omni/projector.py
veomni/models/seed_omni/decoder/__init__.py
veomni/models/seed_omni/decoder/base.py
veomni/models/seed_omni/decoder/cosmos/__init__.py
veomni/models/seed_omni/decoder/cosmos/configuring_cosmos.py
veomni/models/seed_omni/decoder/cosmos/modeling_cosmos.py
veomni/models/seed_omni/decoder/cosmos/processing_cosmos.py
veomni/models/seed_omni/decoder/instruct_pix2pix/__init__.py
veomni/models/seed_omni/decoder/instruct_pix2pix/configuring_instruct_pix2pix.py
veomni/models/seed_omni/decoder/instruct_pix2pix/modeling_instruct_pix2pix.py
veomni/models/seed_omni/decoder/janusvq16/__init__.py
veomni/models/seed_omni/decoder/janusvq16/configuring_janusvq16.py
veomni/models/seed_omni/decoder/janusvq16/modeling_janusvq16.py
veomni/models/seed_omni/decoder/janusvq16/processing_janusvq16.py
veomni/models/seed_omni/decoder/movqgan/__init__.py
veomni/models/seed_omni/decoder/movqgan/configuration_movqgan.py
veomni/models/seed_omni/decoder/movqgan/modeling_movqgan.py
veomni/models/seed_omni/decoder/movqgan/processing_movqgan.py
veomni/models/seed_omni/decoder/ultra_edit/__init__.py
veomni/models/seed_omni/decoder/ultra_edit/configuring_ultra_edit.py
veomni/models/seed_omni/decoder/ultra_edit/modeling_ultra_edit.py
veomni/models/seed_omni/decoder/ultra_edit/processing_ultra_edit.py
veomni/models/seed_omni/encoder/__init__.py
veomni/models/seed_omni/encoder/base.py
veomni/models/seed_omni/encoder/janus_siglip/__init__.py
veomni/models/seed_omni/encoder/janus_siglip/configuration_janus_siglip.py
veomni/models/seed_omni/encoder/janus_siglip/modeling_janus_siglip.py
veomni/models/seed_omni/encoder/janus_siglip/processing_janus_siglip.py
veomni/models/seed_omni/encoder/qwen2_5_omni_audio_model/__init__.py
veomni/models/seed_omni/encoder/qwen2_5_omni_audio_model/configuration_qwen2_5_omni_audio_model.py
veomni/models/seed_omni/encoder/qwen2_5_omni_audio_model/modeling_qwen2_5_omni_audio_model.py
veomni/models/seed_omni/encoder/qwen2_5_omni_audio_model/processing_qwen2_5_omni_audio_model.py
veomni/models/seed_omni/encoder/qwen2_5_omni_vision_model/__init__.py
veomni/models/seed_omni/encoder/qwen2_5_omni_vision_model/configuration_qwen2_5_omni_vision_model.py
veomni/models/seed_omni/encoder/qwen2_5_omni_vision_model/modeling_qwen2_5_omni_vision_model.py
veomni/models/seed_omni/encoder/qwen2_5_omni_vision_model/processing_qwen2_5_omni_vision_model.py
veomni/models/seed_omni/encoder/qwen2_5_vl_vision_model/__init__.py
veomni/models/seed_omni/encoder/qwen2_5_vl_vision_model/configuration_qwen2_5_vl_vision_model.py
veomni/models/seed_omni/encoder/qwen2_5_vl_vision_model/modeling_qwen2_5_vl_vision_model.py
veomni/models/seed_omni/encoder/qwen2_5_vl_vision_model/processing_qwen2_5_vl_vision_model.py
veomni/models/seed_omni/encoder/qwen2_vl_vision_model/__init__.py
veomni/models/seed_omni/encoder/qwen2_vl_vision_model/configuration_qwen2_vl_vision_model.py
veomni/models/seed_omni/encoder/qwen2_vl_vision_model/modeling_qwen2_vl_vision_model.py
veomni/models/seed_omni/encoder/qwen2_vl_vision_model/processing_qwen2_vl_vision_model.py
veomni/models/seed_omni/foundation/__init__.py
veomni/models/seed_omni/foundation/base.py
veomni/models/seed_omni/foundation/janus_foundation/__init__.py
veomni/models/seed_omni/foundation/janus_foundation/configuration_janus_foundation.py
veomni/models/seed_omni/foundation/janus_foundation/modeling_janus_foundation.py
veomni/models/seed_omni/foundation/llama_foundation/__init__.py
veomni/models/seed_omni/foundation/llama_foundation/configuration_llama_foundation.py
veomni/models/seed_omni/foundation/llama_foundation/modeling_llama_foundation.py
veomni/models/seed_omni/foundation/qwen2_5_omni_foundation/__init__.py
veomni/models/seed_omni/foundation/qwen2_5_omni_foundation/configuration_qwen2_5_omni_foundation.py
veomni/models/seed_omni/foundation/qwen2_5_omni_foundation/modeling_qwen2_5_omni_foundation.py
veomni/models/seed_omni/foundation/qwen2_5_vl_foundation/__init__.py
veomni/models/seed_omni/foundation/qwen2_5_vl_foundation/configuration_qwen2_5_vl_foundation.py
veomni/models/seed_omni/foundation/qwen2_5_vl_foundation/modeling_qwen2_5_vl_foundation.py
veomni/models/seed_omni/foundation/qwen2_vl_foundation/__init__.py
veomni/models/seed_omni/foundation/qwen2_vl_foundation/configuration_qwen2_vl_foundation.py
veomni/models/seed_omni/foundation/qwen2_vl_foundation/modeling_qwen2_vl_foundation.py
veomni/models/seed_omni/foundation/qwen3_moe_foundation/__init__.py
veomni/models/seed_omni/foundation/qwen3_moe_foundation/configuration_qwen3_moe_foundation.py
veomni/models/seed_omni/foundation/qwen3_moe_foundation/modeling_qwen3_moe_foundation.py
veomni/models/seed_omni/foundation/seed_oss_foundation/__init__.py
veomni/models/seed_omni/foundation/seed_oss_foundation/configuration_seed_oss_foundation.py
veomni/models/seed_omni/foundation/seed_oss_foundation/modeling_seed_oss_foundation.py
veomni/models/transformers/__init__.py
veomni/models/transformers/deepseek_v3/__init__.py
veomni/models/transformers/deepseek_v3/configuration_deepseek.py
veomni/models/transformers/deepseek_v3/modeling_deepseek.py
veomni/models/transformers/deepseek_v3/parallel_plan.py
veomni/models/transformers/flux/__init__.py
veomni/models/transformers/flux/config_flux.py
veomni/models/transformers/flux/encode_flux.py
veomni/models/transformers/flux/modeling_flux.py
veomni/models/transformers/flux/utils_flux.py
veomni/models/transformers/janus/__init__.py
veomni/models/transformers/janus/configuration_janus.py
veomni/models/transformers/janus/image_processing_janus.py
veomni/models/transformers/janus/modeling_janus.py
veomni/models/transformers/janus/processing_janus.py
veomni/models/transformers/llama/__init__.py
veomni/models/transformers/llama/modeling_llama.py
veomni/models/transformers/movqgan/__init__.py
veomni/models/transformers/movqgan/configuration_movqgan.py
veomni/models/transformers/movqgan/modeling_movqgan.py
veomni/models/transformers/movqgan/processing_movqgan.py
veomni/models/transformers/qwen2/__init__.py
veomni/models/transformers/qwen2/modeling_qwen2.py
veomni/models/transformers/qwen2_5_omni/__init__.py
veomni/models/transformers/qwen2_5_omni/configuration_qwen2_5_omni.py
veomni/models/transformers/qwen2_5_omni/modeling_qwen2_5_omni.py
veomni/models/transformers/qwen2_5_omni/processing_qwen2_5_omni.py
veomni/models/transformers/qwen2_5vl/__init__.py
veomni/models/transformers/qwen2_5vl/modeling_qwen2_5_vl.py
veomni/models/transformers/qwen2_vl/__init__.py
veomni/models/transformers/qwen2_vl/modeling_qwen2_vl.py
veomni/models/transformers/qwen3/__init__.py
veomni/models/transformers/qwen3/modeling_qwen3.py
veomni/models/transformers/qwen3_moe/__init__.py
veomni/models/transformers/qwen3_moe/configuration_qwen3_moe.py
veomni/models/transformers/qwen3_moe/modeling_qwen3_moe.py
veomni/models/transformers/qwen3_moe/parallel_plan.py
veomni/models/transformers/seed_oss/__init__.py
veomni/models/transformers/seed_oss/modeling_seed_oss.py
veomni/models/transformers/wan/__init__.py
veomni/models/transformers/wan/config_wan.py
veomni/models/transformers/wan/modeling_wan.py
veomni/ops/__init__.py
veomni/ops/attention.py
veomni/ops/fused_moe.py
veomni/ops/loss.py
veomni/ops/dit/rope_wan/rotary.py
veomni/ops/group_gemm/__init__.py
veomni/ops/group_gemm/kernel/__init__.py
veomni/ops/group_gemm/kernel/group_gemm.py
veomni/ops/group_gemm/kernel/moe.py
veomni/ops/group_gemm/kernel/triton_utils/__init__.py
veomni/ops/group_gemm/kernel/triton_utils/activation.py
veomni/ops/group_gemm/kernel/triton_utils/memory.py
veomni/ops/group_gemm/kernel/triton_utils/utils.py
veomni/ops/group_gemm/utils/__init__.py
veomni/ops/group_gemm/utils/benchmark_utils.py
veomni/ops/group_gemm/utils/config.py
veomni/ops/group_gemm/utils/device.py
veomni/ops/group_gemm/utils/envvars.py
veomni/ops/group_gemm/utils/kernel.py
veomni/ops/group_gemm/utils/path.py
veomni/ops/group_gemm/utils/pretuned.py
veomni/optim/__init__.py
veomni/optim/lr_scheduler.py
veomni/optim/optimizer.py
veomni/schedulers/flow_match.py
veomni/utils/__init__.py
veomni/utils/arguments.py
veomni/utils/checkpoint_utils.py
veomni/utils/count_flops.py
veomni/utils/device.py
veomni/utils/dist_utils.py
veomni/utils/dit_utils.py
veomni/utils/fs.py
veomni/utils/hdfs_io.py
veomni/utils/helper.py
veomni/utils/import_utils.py
veomni/utils/logging.py
veomni/utils/lora_utils.py
veomni/utils/model_utils.py
veomni/utils/multisource_utils.py
veomni/utils/recompute_utils.py
veomni/utils/seqlen_pos_transform_utils.py