From 7ababf14c7f07b1bacfa98e09dc06fe4e3111818 Mon Sep 17 00:00:00 2001 From: smthemex <138738845+smthemex@users.noreply.github.com> Date: Mon, 24 Feb 2025 11:40:31 +0800 Subject: [PATCH] Create config.json --- .../semantic_ckpts/hf_1_325000/config.json | 71 +++++++++++++++++++ 1 file changed, 71 insertions(+) create mode 100644 inference/xcodec_mini_infer/semantic_ckpts/hf_1_325000/config.json diff --git a/inference/xcodec_mini_infer/semantic_ckpts/hf_1_325000/config.json b/inference/xcodec_mini_infer/semantic_ckpts/hf_1_325000/config.json new file mode 100644 index 0000000..acc9c77 --- /dev/null +++ b/inference/xcodec_mini_infer/semantic_ckpts/hf_1_325000/config.json @@ -0,0 +1,71 @@ +{ + "activation_dropout": 0.1, + "apply_spec_augment": true, + "architectures": [ + "HubertModel" + ], + "attention_dropout": 0.1, + "bos_token_id": 1, + "classifier_proj_size": 256, + "conv_bias": false, + "conv_dim": [ + 512, + 512, + 512, + 512, + 512, + 512, + 512 + ], + "conv_kernel": [ + 10, + 3, + 3, + 3, + 3, + 2, + 2 + ], + "conv_stride": [ + 5, + 2, + 2, + 2, + 2, + 2, + 2 + ], + "ctc_loss_reduction": "sum", + "ctc_zero_infinity": false, + "do_stable_layer_norm": false, + "eos_token_id": 2, + "feat_extract_activation": "gelu", + "feat_extract_norm": "group", + "feat_proj_dropout": 0.0, + "feat_proj_layer_norm": true, + "final_dropout": 0.1, + "hidden_act": "gelu", + "hidden_dropout": 0.1, + "hidden_size": 768, + "initializer_range": 0.02, + "intermediate_size": 3072, + "layer_norm_eps": 1e-05, + "layerdrop": 0.1, + "mask_feature_length": 10, + "mask_feature_min_masks": 0, + "mask_feature_prob": 0.0, + "mask_time_length": 10, + "mask_time_min_masks": 2, + "mask_time_prob": 0.05, + "model_type": "hubert", + "num_attention_heads": 12, + "num_conv_pos_embedding_groups": 16, + "num_conv_pos_embeddings": 128, + "num_feat_extract_layers": 7, + "num_hidden_layers": 12, + "pad_token_id": 0, + "torch_dtype": "float32", + "transformers_version": "4.27.3", + "use_weighted_layer_sum": false, + "vocab_size": 32 +}