LICENSE
README.md
pyproject.toml
docs/conf.py
docs/examples/generate_wan_dataset.py
scripts/ci/check_doc_task_paths.py
scripts/deepseek_v3/fp8_cast_bf16.py
scripts/deepseek_v3/kernel.py
scripts/inference/infer_omni.py
scripts/moe_ckpt_merge/moe_merge.py
scripts/moe_ckpt_merge/moe_split.py
scripts/multimodal/convert_data/tom-and-jerry.py
scripts/profile/merge_chrome_trace.py
tasks/deprecated_task/train_flux.py
tasks/deprecated_task/train_qwen_vl.py
tasks/deprecated_task/train_torch.py
tasks/deprecated_task/train_wan.py
tasks/infer/infer_omni_model.py
tasks/infer/infer_qwen2_vl.py
tasks/infer/infer_text.py
tasks/omni/train_omni_model.py
veomni/__init__.py
veomni/_version.py
veomni.egg-info/PKG-INFO
veomni.egg-info/SOURCES.txt
veomni.egg-info/dependency_links.txt
veomni.egg-info/requires.txt
veomni.egg-info/top_level.txt
veomni/arguments/__init__.py
veomni/arguments/arguments_types.py
veomni/arguments/parser.py
veomni/checkpoint/__init__.py
veomni/checkpoint/checkpointer.py
veomni/checkpoint/dcp_checkpointer.py
veomni/checkpoint/dcp_consolidation.py
veomni/data/__init__.py
veomni/data/chat_template.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/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/data/multimodal/dit/data_transform.py
veomni/data/multimodal/dit/preprocess.py
veomni/distributed/__init__.py
veomni/distributed/checkpoint.py
veomni/distributed/clip_grad_norm.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/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/checkpoint_tensor_loading.py
veomni/models/loader.py
veomni/models/module_utils.py
veomni/models/diffusers/__init__.py
veomni/models/diffusers/qwen_image/__init__.py
veomni/models/diffusers/qwen_image/qwen_image_condition/__init__.py
veomni/models/diffusers/qwen_image/qwen_image_condition/configuration_qwen_image_condition.py
veomni/models/diffusers/qwen_image/qwen_image_condition/modeling_qwen_image_condition.py
veomni/models/diffusers/qwen_image/qwen_image_transformer/__init__.py
veomni/models/diffusers/qwen_image/qwen_image_transformer/configuration_qwen_image_transformer.py
veomni/models/diffusers/qwen_image/qwen_image_transformer/modeling_qwen_image_transformer.py
veomni/models/diffusers/wan_t2v/__init__.py
veomni/models/diffusers/wan_t2v/wan_condition/__init__.py
veomni/models/diffusers/wan_t2v/wan_condition/configuration_wan_condition.py
veomni/models/diffusers/wan_t2v/wan_condition/modeling_wan_condition.py
veomni/models/diffusers/wan_t2v/wan_transformer/__init__.py
veomni/models/diffusers/wan_t2v/wan_transformer/configuration_wan_transformer.py
veomni/models/diffusers/wan_t2v/wan_transformer/modeling_wan_transformer.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/configuration_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/configuration_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/configuration_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/configuration_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/attention_utils.py
veomni/models/transformers/deepseek_v3/__init__.py
veomni/models/transformers/deepseek_v3/checkpoint_tensor_converter.py
veomni/models/transformers/deepseek_v3/deepseek_v3_gpu_patch_gen_config.py
veomni/models/transformers/deepseek_v3/deepseek_v3_npu_patch_gen_config.py
veomni/models/transformers/deepseek_v3/device_patch.py
veomni/models/transformers/deepseek_v3/parallel_plan.py
veomni/models/transformers/deepseek_v3/generated/patched_modeling_deepseek_v3_gpu.py
veomni/models/transformers/deepseek_v3/generated/patched_modeling_deepseek_v3_npu.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/glm_moe_dsa/__init__.py
veomni/models/transformers/glm_moe_dsa/glm_moe_dsa_gpu_patch_gen_config.py
veomni/models/transformers/glm_moe_dsa/glm_moe_dsa_npu_patch_gen_config.py
veomni/models/transformers/glm_moe_dsa/generated/patched_modeling_glm_moe_dsa_gpu.py
veomni/models/transformers/glm_moe_dsa/generated/patched_modeling_glm_moe_dsa_npu.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/llama_gpu_patch_gen_config.py
veomni/models/transformers/llama/generated/patched_modeling_llama_gpu.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/qwen2_gpu_patch_gen_config.py
veomni/models/transformers/qwen2/generated/patched_modeling_qwen2_gpu.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/processing_qwen2_5_omni.py
veomni/models/transformers/qwen2_5_omni/qwen2_5_omni_gpu_patch_gen_config.py
veomni/models/transformers/qwen2_5_omni/generated/patched_modeling_qwen2_5_omni_gpu.py
veomni/models/transformers/qwen2_5vl/__init__.py
veomni/models/transformers/qwen2_5vl/qwen2_5_vl_gpu_patch_gen_config.py
veomni/models/transformers/qwen2_5vl/generated/patched_modeling_qwen2_5_vl_gpu.py
veomni/models/transformers/qwen2_vl/__init__.py
veomni/models/transformers/qwen2_vl/qwen2_vl_gpu_patch_gen_config.py
veomni/models/transformers/qwen2_vl/generated/patched_modeling_qwen2_vl_gpu.py
veomni/models/transformers/qwen3/__init__.py
veomni/models/transformers/qwen3/qwen3_gpu_patch_gen_config.py
veomni/models/transformers/qwen3/qwen3_npu_patch_gen_config.py
veomni/models/transformers/qwen3/generated/patched_modeling_qwen3_gpu.py
veomni/models/transformers/qwen3/generated/patched_modeling_qwen3_npu.py
veomni/models/transformers/qwen3_5/__init__.py
veomni/models/transformers/qwen3_5/qwen3_5_gpu_patch_gen_config.py
veomni/models/transformers/qwen3_5/qwen3_5_npu_patch_gen_config.py
veomni/models/transformers/qwen3_5/generated/patched_modeling_qwen3_5_gpu.py
veomni/models/transformers/qwen3_5/generated/patched_modeling_qwen3_5_npu.py
veomni/models/transformers/qwen3_5_moe/__init__.py
veomni/models/transformers/qwen3_5_moe/parallel_plan.py
veomni/models/transformers/qwen3_5_moe/qwen3_5_moe_gpu_patch_gen_config.py
veomni/models/transformers/qwen3_5_moe/qwen3_5_moe_npu_patch_gen_config.py
veomni/models/transformers/qwen3_5_moe/generated/patched_modeling_qwen3_5_moe_gpu.py
veomni/models/transformers/qwen3_5_moe/generated/patched_modeling_qwen3_5_moe_npu.py
veomni/models/transformers/qwen3_moe/__init__.py
veomni/models/transformers/qwen3_moe/checkpoint_tensor_converter.py
veomni/models/transformers/qwen3_moe/parallel_plan.py
veomni/models/transformers/qwen3_moe/qwen3_moe_gpu_patch_gen_config.py
veomni/models/transformers/qwen3_moe/qwen3_moe_npu_patch_gen_config.py
veomni/models/transformers/qwen3_moe/generated/patched_modeling_qwen3_moe_gpu.py
veomni/models/transformers/qwen3_moe/generated/patched_modeling_qwen3_moe_npu.py
veomni/models/transformers/qwen3_omni_moe/__init__.py
veomni/models/transformers/qwen3_omni_moe/checkpoint_tensor_converter.py
veomni/models/transformers/qwen3_omni_moe/configuration_qwen3_omni_moe.py
veomni/models/transformers/qwen3_omni_moe/parallel_plan.py
veomni/models/transformers/qwen3_omni_moe/processing_qwen3_omni_moe.py
veomni/models/transformers/qwen3_omni_moe/qwen3_omni_moe_gpu_patch_gen_config.py
veomni/models/transformers/qwen3_omni_moe/generated/patched_modeling_qwen3_omni_moe_gpu.py
veomni/models/transformers/qwen3_vl/__init__.py
veomni/models/transformers/qwen3_vl/qwen3_vl_gpu_patch_gen_config.py
veomni/models/transformers/qwen3_vl/qwen3_vl_npu_patch_gen_config.py
veomni/models/transformers/qwen3_vl/generated/patched_modeling_qwen3_vl_gpu.py
veomni/models/transformers/qwen3_vl/generated/patched_modeling_qwen3_vl_npu.py
veomni/models/transformers/qwen3_vl_moe/__init__.py
veomni/models/transformers/qwen3_vl_moe/checkpoint_tensor_converter.py
veomni/models/transformers/qwen3_vl_moe/parallel_plan.py
veomni/models/transformers/qwen3_vl_moe/qwen3_vl_moe_gpu_patch_gen_config.py
veomni/models/transformers/qwen3_vl_moe/qwen3_vl_moe_npu_patch_gen_config.py
veomni/models/transformers/qwen3_vl_moe/generated/patched_modeling_qwen3_vl_moe_gpu.py
veomni/models/transformers/qwen3_vl_moe/generated/patched_modeling_qwen3_vl_moe_npu.py
veomni/models/transformers/seed_oss/__init__.py
veomni/models/transformers/seed_oss/seed_oss_gpu_patch_gen_config.py
veomni/models/transformers/seed_oss/seed_oss_npu_patch_gen_config.py
veomni/models/transformers/seed_oss/generated/patched_modeling_seed_oss_gpu.py
veomni/models/transformers/seed_oss/generated/patched_modeling_seed_oss_npu.py
veomni/models/transformers/wan/__init__.py
veomni/models/transformers/wan/config_wan.py
veomni/models/transformers/wan/device_patch.py
veomni/models/transformers/wan/modeling_wan.py
veomni/models/transformers/wan/npu_patch.py
veomni/ops/__init__.py
veomni/ops/dispatch.py
veomni/ops/kernel_registry.py
veomni/ops/batch_invariant_ops/__init__.py
veomni/ops/batch_invariant_ops/batch_invariant_ops.py
veomni/ops/config/__init__.py
veomni/ops/config/registry.py
veomni/ops/config/singleton.py
veomni/ops/kernels/__init__.py
veomni/ops/kernels/attention/__init__.py
veomni/ops/kernels/cross_entropy/__init__.py
veomni/ops/kernels/cross_entropy/chunk_logprobs.py
veomni/ops/kernels/cross_entropy/chunk_loss.py
veomni/ops/kernels/cross_entropy/chunk_topk_distill.py
veomni/ops/kernels/cross_entropy/eager.py
veomni/ops/kernels/cross_entropy/liger.py
veomni/ops/kernels/gated_delta_rule/__init__.py
veomni/ops/kernels/gated_delta_rule/npu_rms_norm_gated.py
veomni/ops/kernels/load_balancing_loss/__init__.py
veomni/ops/kernels/load_balancing_loss/eager.py
veomni/ops/kernels/load_balancing_loss/triton.py
veomni/ops/kernels/moe/__init__.py
veomni/ops/kernels/moe/group_gemm.py
veomni/ops/kernels/moe/npu_group_gemm.py
veomni/ops/kernels/moe/quack_gemm.py
veomni/ops/kernels/moe/_kernels/__init__.py
veomni/ops/kernels/moe/_kernels/kernel/__init__.py
veomni/ops/kernels/moe/_kernels/kernel/group_gemm.py
veomni/ops/kernels/moe/_kernels/kernel/moe.py
veomni/ops/kernels/moe/_kernels/kernel/npu_group_gemm.py
veomni/ops/kernels/moe/_kernels/kernel/triton_utils/__init__.py
veomni/ops/kernels/moe/_kernels/kernel/triton_utils/activation.py
veomni/ops/kernels/moe/_kernels/kernel/triton_utils/memory.py
veomni/ops/kernels/moe/_kernels/kernel/triton_utils/utils.py
veomni/ops/kernels/moe/_kernels/utils/__init__.py
veomni/ops/kernels/moe/_kernels/utils/benchmark_utils.py
veomni/ops/kernels/moe/_kernels/utils/config.py
veomni/ops/kernels/moe/_kernels/utils/device.py
veomni/ops/kernels/moe/_kernels/utils/envvars.py
veomni/ops/kernels/moe/_kernels/utils/kernel.py
veomni/ops/kernels/moe/_kernels/utils/path.py
veomni/ops/kernels/moe/_kernels/utils/pretuned.py
veomni/ops/kernels/rms_norm/__init__.py
veomni/ops/kernels/rms_norm/npu.py
veomni/ops/kernels/rms_norm/triton_batch_invariant.py
veomni/ops/kernels/rotary/__init__.py
veomni/ops/kernels/rotary/npu.py
veomni/ops/kernels/rotary/triton_deterministic.py
veomni/ops/kernels/rotary/triton_wan.py
veomni/ops/kernels/swiglu/__init__.py
veomni/ops/liger/__init__.py
veomni/ops/platform/__init__.py
veomni/ops/platform/npu/__init__.py
veomni/ops/platform/npu/hccl_premul_sum.py
veomni/optim/__init__.py
veomni/optim/lr_scheduler.py
veomni/optim/muon.py
veomni/optim/optimizer.py
veomni/patchgen/__init__.py
veomni/patchgen/check_patchgen.py
veomni/patchgen/codegen.py
veomni/patchgen/patch_spec.py
veomni/patchgen/run_codegen.py
veomni/schedulers/flow_match.py
veomni/trainer/__init__.py
veomni/trainer/base.py
veomni/trainer/base_rl_trainer.py
veomni/trainer/dit_trainer.py
veomni/trainer/text_dpo_trainer.py
veomni/trainer/text_trainer.py
veomni/trainer/vlm_trainer.py
veomni/trainer/callbacks/__init__.py
veomni/trainer/callbacks/base.py
veomni/trainer/callbacks/checkpoint_callback.py
veomni/trainer/callbacks/evaluate_callback.py
veomni/trainer/callbacks/trace_callback.py
veomni/utils/__init__.py
veomni/utils/checkpoint_utils.py
veomni/utils/constants.py
veomni/utils/count_flops.py
veomni/utils/device.py
veomni/utils/dist_utils.py
veomni/utils/dit_utils.py
veomni/utils/env.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/loss_utils.py
veomni/utils/model_outputs.py
veomni/utils/model_utils.py
veomni/utils/moe_monitor.py
veomni/utils/moe_router_replay.py
veomni/utils/multisource_utils.py
veomni/utils/recompute_utils.py
veomni/utils/registry.py
veomni/utils/save_safetensor_utils.py
veomni/utils/seqlen_pos_transform_utils.py
veomni/utils/data_balance/balance_sorting_algo.py
veomni/utils/data_balance/data_balance.py