{ "model": "model", "model_type": "qwen3_omni", "model_revision": null, "task_type": "causal_lm", "torch_dtype": "bfloat16", "attn_impl": null, "new_special_tokens": [], "num_labels": null, "problem_type": "single_label_classification", "rope_scaling": { "interleaved": true, "mrope_interleaved": true, "mrope_section": [ 24, 20, 20 ], "rope_type": "default", "type": "default" }, "device_map": null, "max_memory": {}, "max_model_len": null, "local_repo_path": null, "init_strategy": null, "template": "qwen3_omni", "system": null, "max_length": 3072, "truncation_strategy": "delete", "max_pixels": null, "agent_template": null, "norm_bbox": null, "use_chat_template": true, "padding_free": true, "padding_side": "right", "loss_scale": "default", "sequence_parallel_size": 1, "response_prefix": null, "template_backend": "swift", "dataset": [ "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010583.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010584.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010585.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010586.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010587.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010588.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010589.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010590.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010591.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010592.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010593.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010594.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010595.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010596.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010597.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010598.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010599.parquet" ], "val_dataset": [], "split_dataset_ratio": 0.01, "data_seed": 42, "dataset_num_proc": 128, "load_from_cache_file": false, "dataset_shuffle": true, "val_dataset_shuffle": false, "streaming": false, "interleave_prob": null, "stopping_strategy": "first_exhausted", "shuffle_buffer_size": 1000, "download_mode": "reuse_dataset_if_exists", "columns": {}, "strict": false, "remove_unused_columns": true, "model_name": null, "model_author": null, "custom_dataset_info": [], "quant_method": null, "quant_bits": null, "hqq_axis": null, "bnb_4bit_compute_dtype": "bfloat16", "bnb_4bit_quant_type": "nf4", "bnb_4bit_use_double_quant": true, "bnb_4bit_quant_storage": null, "max_new_tokens": null, "temperature": null, "top_k": null, "top_p": null, "repetition_penalty": null, "num_beams": 1, "stream": false, "stop_words": [], "logprobs": false, "top_logprobs": null, "ckpt_dir": "model", "lora_modules": [], "tuner_backend": "peft", "train_type": "full", "adapters": [], "external_plugins": [], "seed": 42, "model_kwargs": {}, "load_args": false, "load_data_args": false, "packing": false, "packing_length": null, "lazy_tokenize": true, "cached_dataset": [], "custom_register_path": [], "use_hf": false, "hub_token": null, "ddp_timeout": 18000000, "ddp_backend": null, "ignore_args_error": false, "use_swift_lora": false, "freeze_llm": false, "freeze_vit": false, "freeze_aligner": true, "freeze_parameters": [], "freeze_parameters_regex": null, "freeze_parameters_ratio": 0.0, "trainable_parameters": [], "trainable_parameters_regex": null, "adapter_load": null, "target_modules": [ "all-linear" ], "target_regex": null, "modules_to_save": [], "lora_rank": 8, "lora_alpha": 32, "lora_dropout": 0.05, "lora_bias": "none", "lora_dtype": null, "use_rslora": false, "ref_load": null, "ref_adapter_load": null, "beta": 0.1, "rpo_alpha": null, "reference_free": false, "label_smoothing": 0.0, "f_divergence_type": "reverse_kl", "loss_type": "sigmoid", "padded_vocab_size": 152064, "initialize_embedding": false, "mlp_padding_free": false, "dataloader_persistent_workers": true, "dataloader_prefetch_factor": 10, "architectures": "Qwen3OmniMoeForConditionalGeneration", "llm_architectures": null, "max_epochs": null, "enable_dft_loss": false, "enable_channel_loss": false, "original_max_position_embeddings": null, "partial_rotary_factor": null, "use_shared_expert_gate": false, "vit_gradient_checkpointing": true, "gradient_checkpointing_kwargs": null, "linear_num_value_heads": null, "linear_num_key_heads": null, "linear_key_head_dim": null, "linear_value_head_dim": null, "linear_conv_kernel_dim": null, "layer_types": null, "mrope_interleaved": true, "micro_batch_size": 32, "global_batch_size": 256, "recompute_granularity": "selective", "recompute_method": null, "recompute_num_layers": null, "recompute_modules": [ "core_attn", "moe" ], "use_cpu_initialization": false, "deterministic_mode": false, "train_iters": 312, "log_interval": 10, "tensorboard_dir": "/srv/chk/AniMUL-production-next/v177-20260131-191635/runs", "no_masked_softmax_fusion": false, "no_bias_dropout_fusion": false, "no_bias_swiglu_fusion": false, "no_rope_fusion": false, "no_gradient_accumulation_fusion": false, "cross_entropy_loss_fusion": true, "cross_entropy_fusion_impl": "native", "calculate_per_token_loss": true, "use_flash_attn": false, "attention_backend": "flash", "optimizer": "adam", "optimizer_cpu_offload": false, "optimizer_offload_fraction": 1.0, "use_precision_aware_optimizer": true, "main_grads_dtype": "fp32", "main_params_dtype": "fp32", "exp_avg_dtype": "fp32", "exp_avg_sq_dtype": "fp32", "dataloader_type": "cyclic", "manual_gc": false, "manual_gc_interval": 0, "lr": 3e-05, "lr_decay_style": "cosine", "lr_decay_iters": null, "lr_warmup_iters": 0, "lr_warmup_fraction": 0.1, "min_lr": 1.5e-06, "weight_decay": 0.1, "clip_grad": 1.0, "adam_beta1": 0.9, "adam_beta2": 0.95, "adam_eps": 1e-08, "sgd_momentum": 0.9, "save": "/srv/chk/AniMUL-production-next/v177-20260131-191635", "save_interval": 500, "no_save_optim": false, "no_save_rng": false, "load": "/srv/chk/AniMUL-production-next/v176-20260131-125439", "no_load_optim": false, "no_load_rng": false, "finetune": true, "ckpt_format": "torch_dist", "no_initialization": true, "auto_detect_ckpt_format": true, "exit_on_missing_checkpoint": true, "distributed_backend": "nccl", "local_rank": 0, "use_distributed_optimizer": true, "tensor_model_parallel_size": 4, "pipeline_model_parallel_size": 1, "decoder_first_pipeline_num_layers": null, "decoder_last_pipeline_num_layers": null, "sequence_parallel": true, "context_parallel_size": 1, "tp_comm_overlap": false, "overlap_grad_reduce": true, "overlap_param_gather": true, "distributed_timeout_minutes": 300000, "num_layers_per_virtual_pipeline_stage": null, "num_virtual_stages_per_pipeline_rank": null, "microbatch_group_size_per_virtual_pipeline_stage": null, "pipeline_model_parallel_layout": null, "num_layers": 48, "hidden_size": 2048, "ffn_hidden_size": null, "num_attention_heads": 32, "group_query_attention": true, "num_query_groups": 4, "max_position_embeddings": 65536, "position_embedding_type": "mrope", "mrope_section": [ 24, 20, 20 ], "rotary_base": 1000000, "rotary_percent": 1.0, "rotary_interleaved": false, "normalization": "RMSNorm", "norm_epsilon": 1e-06, "swiglu": true, "untie_embeddings_and_output_weights": true, "disable_bias_linear": true, "add_qkv_bias": false, "attention_dropout": 0.0, "hidden_dropout": 0.0, "kv_channels": 128, "qk_layernorm": true, "transformer_impl": "transformer_engine", "num_experts": 128, "moe_layer_freq": "1", "moe_ffn_hidden_size": 768, "moe_shared_expert_intermediate_size": null, "moe_router_topk": 8, "moe_router_pre_softmax": false, "moe_router_dtype": "fp32", "moe_router_score_function": "softmax", "moe_router_bias_update_rate": 0.001, "moe_router_enable_expert_bias": false, "moe_router_topk_scaling_factor": null, "moe_router_load_balancing_type": "aux_loss", "expert_model_parallel_size": 2, "expert_tensor_parallel_size": 1, "moe_token_dispatcher_type": null, "moe_enable_deepep": false, "moe_grouped_gemm": true, "moe_permute_fusion": true, "moe_aux_loss_coeff": 0.001, "moe_z_loss_coeff": null, "moe_shared_expert_overlap": true, "moe_layer_recompute": false, "moe_expert_capacity_factor": null, "moe_pad_expert_input_to_capacity": false, "moe_token_drop_policy": null, "multi_latent_attention": false, "q_lora_rank": null, "kv_lora_rank": 32, "qk_head_dim": 128, "qk_pos_emb_head_dim": 64, "fp8_format": "hybrid", "fp8_recipe": "delayed", "fp8_amax_history_len": 1024, "fp8_amax_compute_algo": "max", "fp8_param_gather": false, "fp16": false, "bf16": true, "apply_query_key_layer_scaling": false, "attention_softmax_in_fp32": true, "log_params_norm": false, "log_throughput": false, "tensorboard_log_interval": 1, "tensorboard_queue_size": 50, "log_timers_to_tensorboard": true, "no_log_learning_rate_to_tensorboard": false, "log_validation_ppl_to_tensorboard": true, "log_memory_to_tensorboard": true, "logging_level": null, "wandb_project": null, "wandb_exp_name": null, "wandb_save_dir": null, "eval_iters": -1, "eval_interval": 500, "seq_length": 3072, "num_workers": 4, "extra_megatron_kwargs": {}, "add_version": true, "rank": 0, "global_world_size": 8, "local_world_size": 8, "model_suffix": "model", "model_info": "ModelInfo(model_type='qwen3_omni', model_dir='/home/jebba/devel/deepcrayon/AniMUL/model', torch_dtype=torch.bfloat16, max_model_len=65536, quant_method=None, quant_bits=None, rope_scaling={'interleaved': True, 'mrope_section': [24, 20, 20], 'rope_type': 'default', 'type': 'default'}, is_moe_model=True, config=None, task_type='causal_lm', num_labels=None)", "model_meta": "ModelMeta(model_type='qwen3_omni', model_groups=[ModelGroup(models=[Model(ms_model_id='Qwen/Qwen3-Omni-30B-A3B-Instruct', hf_model_id='Qwen/Qwen3-Omni-30B-A3B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen3-Omni-30B-A3B-Thinking', hf_model_id='Qwen/Qwen3-Omni-30B-A3B-Thinking', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen3-Omni-30B-A3B-Captioner', hf_model_id='Qwen/Qwen3-Omni-30B-A3B-Captioner', model_path=None, ms_revision=None, hf_revision=None)], ignore_patterns=None, requires=None, tags=[])], template='qwen3_omni', get_function=, model_arch=MultiModelKeys(arch_name='qwen3_omni', embedding=None, module_list=None, lm_head=None, q_proj=None, k_proj=None, v_proj=None, o_proj=None, attention=None, mlp=None, down_proj=None, qkv_proj=None, qk_proj=None, qa_proj=None, qb_proj=None, kv_proj=None, kva_proj=None, kvb_proj=None, language_model=['thinker.model'], aligner=['thinker.audio_tower.proj1', 'thinker.audio_tower.proj2', 'thinker.visual.merger', 'thinker.visual.merger_list'], vision_tower=['thinker.audio_tower', 'thinker.visual'], generator=['talker', 'token2wav']), architectures=['Qwen3OmniMoeForConditionalGeneration'], additional_saved_files=[], torch_dtype=None, is_multimodal=True, is_reward=False, task_type=None, ignore_patterns=None, requires=['transformers>=4.57.dev0', 'soundfile', 'decord', 'qwen_omni_utils'], tags=['vision', 'video', 'audio'])", "model_dir": "/home/jebba/devel/deepcrayon/AniMUL/model", "hub": "", "megatron_model_meta": "MMGPTMegatronModelMeta(megatron_model_type='qwen3_omni', model_types=['qwen3_omni'], convert_mcore2hf=, convert_hf2mcore=, model_cls=, convert_hf_config=, get_transformer_layer_spec=None, model_provider=, visual_cls=, extra_args_provider=None)", "extra_args": { "model": "model", "model_type": "qwen3_omni", "model_revision": null, "task_type": "causal_lm", "torch_dtype": "bfloat16", "attn_impl": null, "new_special_tokens": [], "num_labels": null, "problem_type": "single_label_classification", "rope_scaling": { "interleaved": true, "mrope_interleaved": true, "mrope_section": [ 24, 20, 20 ], "rope_type": "default", "type": "default" }, "device_map": null, "max_memory": {}, "max_model_len": null, "local_repo_path": null, "init_strategy": null, "template": "qwen3_omni", "system": null, "max_length": 3072, "truncation_strategy": "delete", "max_pixels": null, "agent_template": null, "norm_bbox": null, "use_chat_template": true, "padding_free": true, "padding_side": "right", "sequence_parallel_size": 1, "response_prefix": null, "template_backend": "swift", "dataset": [ "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010583.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010584.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010585.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010586.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010587.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010588.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010589.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010590.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010591.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010592.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010593.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010594.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010595.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010596.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010597.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010598.parquet", "/srv/hf/datasets/EarthSpeciesProject/NatureLM-audio-training/train/part3/shard_010599.parquet" ], "val_dataset": [], "split_dataset_ratio": 0.01, "data_seed": 42, "dataset_num_proc": 128, "load_from_cache_file": false, "dataset_shuffle": true, "val_dataset_shuffle": false, "streaming": false, "interleave_prob": null, "stopping_strategy": "first_exhausted", "shuffle_buffer_size": 1000, "download_mode": "reuse_dataset_if_exists", "columns": {}, "strict": false, "remove_unused_columns": true, "model_name": null, "model_author": null, "custom_dataset_info": [], "quant_method": null, "quant_bits": null, "hqq_axis": null, "bnb_4bit_compute_dtype": "bfloat16", "bnb_4bit_quant_type": "nf4", "bnb_4bit_use_double_quant": true, "bnb_4bit_quant_storage": null, "max_new_tokens": null, "temperature": null, "top_k": null, "top_p": null, "repetition_penalty": null, "num_beams": 1, "stream": false, "stop_words": [], "logprobs": false, "top_logprobs": null, "ckpt_dir": "model", "lora_modules": [], "tuner_backend": "peft", "train_type": "full", "adapters": [], "external_plugins": [], "model_kwargs": {}, "load_args": false, "load_data_args": false, "packing": false, "packing_length": null, "lazy_tokenize": true, "cached_dataset": [], "custom_register_path": [], "use_hf": false, "hub_token": null, "ddp_timeout": 18000000, "ddp_backend": null, "ignore_args_error": false, "use_swift_lora": false, "freeze_llm": false, "freeze_vit": false, "freeze_aligner": true, "freeze_parameters": [], "freeze_parameters_regex": null, "freeze_parameters_ratio": 0.0, "trainable_parameters": [], "trainable_parameters_regex": null, "adapter_load": null, "target_modules": [ "all-linear" ], "target_regex": null, "modules_to_save": [], "lora_rank": 8, "lora_alpha": 32, "lora_dropout": 0.05, "lora_bias": "none", "lora_dtype": null, "use_rslora": false, "ref_load": null, "ref_adapter_load": null, "beta": 0.1, "rpo_alpha": null, "reference_free": false, "label_smoothing": 0.0, "f_divergence_type": "reverse_kl", "loss_type": "sigmoid", "padded_vocab_size": 152064, "initialize_embedding": false, "mlp_padding_free": false, "dataloader_persistent_workers": true, "dataloader_prefetch_factor": 10, "architectures": "Qwen3OmniMoeForConditionalGeneration", "llm_architectures": null, "max_epochs": null, "enable_dft_loss": false, "enable_channel_loss": false, "original_max_position_embeddings": null, "partial_rotary_factor": null, "use_shared_expert_gate": false, "vit_gradient_checkpointing": true, "gradient_checkpointing_kwargs": null, "linear_num_value_heads": null, "linear_num_key_heads": null, "linear_key_head_dim": null, "linear_value_head_dim": null, "linear_conv_kernel_dim": null, "layer_types": null, "mrope_interleaved": true, "add_version": true, "model_info": "ModelInfo(model_type='qwen3_omni', model_dir='/home/jebba/devel/deepcrayon/AniMUL/model', torch_dtype=torch.bfloat16, max_model_len=65536, quant_method=None, quant_bits=None, rope_scaling={'interleaved': True, 'mrope_section': [24, 20, 20], 'rope_type': 'default', 'type': 'default'}, is_moe_model=True, config=None, task_type='causal_lm', num_labels=None)", "model_meta": "ModelMeta(model_type='qwen3_omni', model_groups=[ModelGroup(models=[Model(ms_model_id='Qwen/Qwen3-Omni-30B-A3B-Instruct', hf_model_id='Qwen/Qwen3-Omni-30B-A3B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen3-Omni-30B-A3B-Thinking', hf_model_id='Qwen/Qwen3-Omni-30B-A3B-Thinking', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen3-Omni-30B-A3B-Captioner', hf_model_id='Qwen/Qwen3-Omni-30B-A3B-Captioner', model_path=None, ms_revision=None, hf_revision=None)], ignore_patterns=None, requires=None, tags=[])], template='qwen3_omni', get_function=, model_arch=MultiModelKeys(arch_name='qwen3_omni', embedding=None, module_list=None, lm_head=None, q_proj=None, k_proj=None, v_proj=None, o_proj=None, attention=None, mlp=None, down_proj=None, qkv_proj=None, qk_proj=None, qa_proj=None, qb_proj=None, kv_proj=None, kva_proj=None, kvb_proj=None, language_model=['thinker.model'], aligner=['thinker.audio_tower.proj1', 'thinker.audio_tower.proj2', 'thinker.visual.merger', 'thinker.visual.merger_list'], vision_tower=['thinker.audio_tower', 'thinker.visual'], generator=['talker', 'token2wav']), architectures=['Qwen3OmniMoeForConditionalGeneration'], additional_saved_files=[], torch_dtype=None, is_multimodal=True, is_reward=False, task_type=None, ignore_patterns=None, requires=['transformers>=4.57.dev0', 'soundfile', 'decord', 'qwen_omni_utils'], tags=['vision', 'video', 'audio'])", "megatron_model_meta": "MMGPTMegatronModelMeta(megatron_model_type='qwen3_omni', model_types=['qwen3_omni'], convert_mcore2hf=, convert_hf2mcore=, model_cls=, convert_hf_config=, get_transformer_layer_spec=None, model_provider=, visual_cls=, extra_args_provider=None)" } }