| { | |
| "architectures": [ | |
| "Mammothmoda2Model" | |
| ], | |
| "gen_axes_dim_rope": [ | |
| 40, | |
| 40, | |
| 40 | |
| ], | |
| "gen_axes_lens": [ | |
| 10000, | |
| 10000, | |
| 10000 | |
| ], | |
| "gen_condition_mode": "text_image", | |
| "gen_dit_config": { | |
| "_class_name": "OmniGen2Transformer2DModel", | |
| "_diffusers_version": "0.33.1", | |
| "axes_dim_rope": [ | |
| 40, | |
| 40, | |
| 40 | |
| ], | |
| "axes_lens": [ | |
| 1024, | |
| 1664, | |
| 1664 | |
| ], | |
| "ffn_dim_multiplier": null, | |
| "hidden_size": 2520, | |
| "in_channels": 16, | |
| "multiple_of": 256, | |
| "norm_eps": 1e-05, | |
| "num_attention_heads": 21, | |
| "num_kv_heads": 7, | |
| "num_layers": 16, | |
| "num_refiner_layers": 2, | |
| "out_channels": null, | |
| "patch_size": 2, | |
| "text_feat_dim": 2048, | |
| "timestep_scale": 1000.0 | |
| }, | |
| "gen_image_condition_refiner_config": { | |
| "num_layers": 2, | |
| "num_queries": 128 | |
| }, | |
| "gen_transport_config": { | |
| "do_shift": true, | |
| "seq_len": 4096, | |
| "snr_type": "lognorm" | |
| }, | |
| "gen_vae_config": { | |
| "_class_name": "AutoencoderKL", | |
| "_diffusers_version": "0.30.0.dev0", | |
| "_name_or_path": "../checkpoints/flux-dev", | |
| "act_fn": "silu", | |
| "block_out_channels": [ | |
| 128, | |
| 256, | |
| 512, | |
| 512 | |
| ], | |
| "down_block_types": [ | |
| "DownEncoderBlock2D", | |
| "DownEncoderBlock2D", | |
| "DownEncoderBlock2D", | |
| "DownEncoderBlock2D" | |
| ], | |
| "force_upcast": true, | |
| "in_channels": 3, | |
| "latent_channels": 16, | |
| "latents_mean": null, | |
| "latents_std": null, | |
| "layers_per_block": 2, | |
| "mid_block_add_attention": true, | |
| "norm_num_groups": 32, | |
| "out_channels": 3, | |
| "sample_size": 1024, | |
| "scaling_factor": 0.3611, | |
| "shift_factor": 0.1159, | |
| "up_block_types": [ | |
| "UpDecoderBlock2D", | |
| "UpDecoderBlock2D", | |
| "UpDecoderBlock2D", | |
| "UpDecoderBlock2D" | |
| ], | |
| "use_post_quant_conv": false, | |
| "use_quant_conv": false | |
| }, | |
| "initializer_range": 0.02, | |
| "llm_config": { | |
| "attention_dropout": 0.0, | |
| "bos_token_id": 151643, | |
| "eos_token_id": 151645, | |
| "extra_gen_vocab": true, | |
| "gen_vocab_size": 32800, | |
| "gen_vocab_start_index": 152064, | |
| "hidden_act": "silu", | |
| "hidden_size": 3584, | |
| "image_token_id": 151655, | |
| "initializer_range": 0.02, | |
| "intermediate_size": 18944, | |
| "max_position_embeddings": 128000, | |
| "max_window_layers": 28, | |
| "model_type": "mammothmoda2_qwen2_5_vl", | |
| "moe_type": "ffn-14:28", | |
| "num_attention_heads": 28, | |
| "num_hidden_layers": 28, | |
| "num_key_value_heads": 4, | |
| "rms_norm_eps": 1e-06, | |
| "rope_scaling": { | |
| "mrope_section": [ | |
| 16, | |
| 24, | |
| 24 | |
| ], | |
| "rope_type": "default", | |
| "type": "default" | |
| }, | |
| "rope_theta": 1000000.0, | |
| "sliding_window": 32768, | |
| "text_config": { | |
| "_name_or_path": "", | |
| "add_cross_attention": false, | |
| "architectures": [ | |
| "Qwen2_5_VLForConditionalGeneration" | |
| ], | |
| "attention_dropout": 0.0, | |
| "bad_words_ids": null, | |
| "begin_suppress_tokens": null, | |
| "bos_token_id": 151643, | |
| "chunk_size_feed_forward": 0, | |
| "cross_attention_hidden_size": null, | |
| "decoder_start_token_id": null, | |
| "diversity_penalty": 0.0, | |
| "do_sample": false, | |
| "early_stopping": false, | |
| "encoder_no_repeat_ngram_size": 0, | |
| "eos_token_id": 151645, | |
| "exponential_decay_length_penalty": null, | |
| "extra_gen_vocab": true, | |
| "finetuning_task": null, | |
| "forced_bos_token_id": null, | |
| "forced_eos_token_id": null, | |
| "gen_vocab_size": 32800, | |
| "gen_vocab_start_index": 152064, | |
| "hidden_act": "silu", | |
| "hidden_size": 3584, | |
| "id2label": { | |
| "0": "LABEL_0", | |
| "1": "LABEL_1" | |
| }, | |
| "image_token_id": null, | |
| "initializer_range": 0.02, | |
| "intermediate_size": 18944, | |
| "is_decoder": false, | |
| "is_encoder_decoder": false, | |
| "label2id": { | |
| "LABEL_0": 0, | |
| "LABEL_1": 1 | |
| }, | |
| "layer_types": [ | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention", | |
| "full_attention" | |
| ], | |
| "length_penalty": 1.0, | |
| "max_length": 20, | |
| "max_position_embeddings": 128000, | |
| "max_window_layers": 28, | |
| "min_length": 0, | |
| "model_type": "mammothmoda2_qwen2_5_vl_text", | |
| "moe_type": "ffn-14:28", | |
| "no_repeat_ngram_size": 0, | |
| "num_attention_heads": 28, | |
| "num_beam_groups": 1, | |
| "num_beams": 1, | |
| "num_hidden_layers": 28, | |
| "num_key_value_heads": 4, | |
| "num_return_sequences": 1, | |
| "output_attentions": false, | |
| "output_hidden_states": false, | |
| "output_scores": false, | |
| "pad_token_id": null, | |
| "prefix": null, | |
| "problem_type": null, | |
| "pruned_heads": {}, | |
| "remove_invalid_values": false, | |
| "repetition_penalty": 1.0, | |
| "return_dict": true, | |
| "return_dict_in_generate": false, | |
| "rms_norm_eps": 1e-06, | |
| "rope_scaling": { | |
| "mrope_section": [ | |
| 16, | |
| 24, | |
| 24 | |
| ], | |
| "rope_type": "default", | |
| "type": "default" | |
| }, | |
| "rope_theta": 1000000.0, | |
| "sep_token_id": null, | |
| "sliding_window": null, | |
| "suppress_tokens": null, | |
| "task_specific_params": null, | |
| "temperature": 1.0, | |
| "tf_legacy_loss": false, | |
| "tie_encoder_decoder": false, | |
| "tie_word_embeddings": false, | |
| "tokenizer_class": null, | |
| "top_k": 50, | |
| "top_p": 1.0, | |
| "torch_dtype": "bfloat16", | |
| "torchscript": false, | |
| "typical_p": 1.0, | |
| "unpack_forward": true, | |
| "use_bfloat16": false, | |
| "use_cache": true, | |
| "use_sliding_window": false, | |
| "video_token_id": null, | |
| "vision_end_token_id": 151653, | |
| "vision_start_token_id": 151652, | |
| "vision_token_id": 151654, | |
| "vocab_size": 152064 | |
| }, | |
| "tie_word_embeddings": false, | |
| "torch_dtype": "float32", | |
| "unpack_forward": true, | |
| "use_cache": true, | |
| "use_sliding_window": false, | |
| "video_token_id": 151656, | |
| "vision_config": { | |
| "_name_or_path": "", | |
| "add_cross_attention": false, | |
| "architectures": null, | |
| "bad_words_ids": null, | |
| "begin_suppress_tokens": null, | |
| "bos_token_id": null, | |
| "chunk_size_feed_forward": 0, | |
| "cross_attention_hidden_size": null, | |
| "decoder_start_token_id": null, | |
| "depth": 32, | |
| "diversity_penalty": 0.0, | |
| "do_sample": false, | |
| "early_stopping": false, | |
| "encoder_no_repeat_ngram_size": 0, | |
| "eos_token_id": null, | |
| "exponential_decay_length_penalty": null, | |
| "finetuning_task": null, | |
| "forced_bos_token_id": null, | |
| "forced_eos_token_id": null, | |
| "fullatt_block_indexes": [ | |
| 7, | |
| 15, | |
| 23, | |
| 31 | |
| ], | |
| "hidden_act": "silu", | |
| "hidden_size": 1280, | |
| "id2label": { | |
| "0": "LABEL_0", | |
| "1": "LABEL_1" | |
| }, | |
| "in_channels": 3, | |
| "in_chans": 3, | |
| "initializer_range": 0.02, | |
| "intermediate_size": 3420, | |
| "is_decoder": false, | |
| "is_encoder_decoder": false, | |
| "label2id": { | |
| "LABEL_0": 0, | |
| "LABEL_1": 1 | |
| }, | |
| "length_penalty": 1.0, | |
| "max_length": 20, | |
| "min_length": 0, | |
| "model_type": "mammothmoda2_qwen2_5_vl", | |
| "no_repeat_ngram_size": 0, | |
| "num_beam_groups": 1, | |
| "num_beams": 1, | |
| "num_heads": 16, | |
| "num_return_sequences": 1, | |
| "out_hidden_size": 3584, | |
| "output_attentions": false, | |
| "output_hidden_states": false, | |
| "output_scores": false, | |
| "pad_token_id": null, | |
| "patch_size": 14, | |
| "prefix": null, | |
| "problem_type": null, | |
| "pruned_heads": {}, | |
| "remove_invalid_values": false, | |
| "repetition_penalty": 1.0, | |
| "return_dict": true, | |
| "return_dict_in_generate": false, | |
| "sep_token_id": null, | |
| "spatial_merge_size": 2, | |
| "spatial_patch_size": 14, | |
| "suppress_tokens": null, | |
| "task_specific_params": null, | |
| "temperature": 1.0, | |
| "temporal_patch_size": 2, | |
| "tf_legacy_loss": false, | |
| "tie_encoder_decoder": false, | |
| "tie_word_embeddings": true, | |
| "tokenizer_class": null, | |
| "tokens_per_second": 2, | |
| "top_k": 50, | |
| "top_p": 1.0, | |
| "torch_dtype": null, | |
| "torchscript": false, | |
| "typical_p": 1.0, | |
| "use_bfloat16": false, | |
| "window_size": 112 | |
| }, | |
| "vision_end_token_id": 151653, | |
| "vision_start_token_id": 151652, | |
| "vision_token_id": 151654, | |
| "vocab_size": 152064 | |
| }, | |
| "llm_moe_init_gen_from_und": true, | |
| "llm_pretrained_path": null, | |
| "model_type": "mammothmoda2", | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.53.2", | |
| "unpack_forward": true, | |
| "verbose_interval": 200, | |
| "vis_freeze": true | |
| } | |