MoLGE Checkpoints

Released weights for MoLGE: Mixture of Language Group Experts.

Download into the repository checkpoints/ directory so training and inference paths match the code:

hf download Sanghyang00/molge-checkpoints --local-dir checkpoints

or:

bash scripts/download_checkpoints.sh

Layout

checkpoints/
  lamaut/          # romanized CTC (LAMA-UT)
  omnictc/         # grapheme CTC (OmniCTC)
  pretrained/      # OmniASR encoder + first-stage CTC full-finetune

Released models

Path Config Notes
lamaut/molge_geographic.pt configs/lamaut/molge_geographic.yaml MoLGE, geographic groups
lamaut/molge_genetic.pt configs/lamaut/molge_genetic.yaml MoLGE, genetic groups
lamaut/molge_lid.pt configs/lamaut/molge_lid.yaml MoLGE, LID groups
lamaut/molge_mlm.pt configs/lamaut/molge_mlm.yaml MoLGE, MLM groups
lamaut/molge_random.pt configs/lamaut/molge_geographic.yaml random grouping ablation
lamaut/mole_frame_{8,16,32}.pt configs/lamaut/mole_frame_*.yaml frame-wise MoLE
lamaut/mole_sample_{8,16,32}.pt configs/lamaut/mole_sample_*.yaml sample-wise MoLE
lamaut/lora_activate.pt configs/lamaut/lora_activate.yaml LoRA baseline
lamaut/lora_total.pt configs/lamaut/lora_total.yaml LoRA baseline
omnictc/... configs/omnictc/... same recipes with --omnictc
omnictc/molge_geographic_noshare.pt configs/omnictc/molge_geographic_noshare.yaml no shared expert
pretrained/omniasr-300m.pt encoder init in YAML converted OmniASR-300M
pretrained/omniasr-1b.pt prototype extraction converted OmniASR-1B
pretrained/omniasr_ctc_fullft_roman.pt --first_stage_ckpt_path LAMA-UT stage-1
pretrained/omniasr_ctc_fullft_original.pt --first_stage_ckpt_path OmniCTC stage-1

Inference

python inference.py \
  --config_path configs/lamaut/molge_geographic.yaml \
  --ckpt_path checkpoints/lamaut/molge_geographic.pt \
  --group_config_path configs/language_groups/geographic.json \
  --data_dir csvs

python inference.py \
  --config_path configs/omnictc/molge_geographic.yaml \
  --ckpt_path checkpoints/omnictc/molge_geographic.pt \
  --group_config_path configs/language_groups/geographic.json \
  --data_dir csvs_omni \
  --omnictc
Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support