Text Generation
Safetensors
GGUF
Japanese
conversational

hou-rai (蓬莱) a150m 7.8b Trainig codes | Other

logo image

~ a model for demo 2026: Qwen3moe architecture 150m * 52 expert 7.8b ~

Model Details

base model: Qwen/Qwen3-0.6B (to slice 150m), to qwen3moe architecture use mergekit-moe

file tree

# ==================================================
./scripts
  ./merge_150m_step1_distill.py
  ./merge_150m_step2_wikipedia.py
  ./merge_150m_step3_aozora.py
  ./merge_150m_step3_aozora_fix.py

  ./train_peft_150m_step1_distll.py
  ./train_peft_150m_step2_wikipedia.py
  ./train_peft_150m_step3_aozora.py
  ./train_peft_150m_step3_aozora_fix.py

# ==================================================
# datasets
# ==================================================
./datasets
  ./Japanese-Wikipedia-202506
    ./
  ./aozorabunko-clean/aozorabunko-dedupe-clean.jsonl.gz
  ./Zero_SFT_Ja_v3.5_Reasoning/answers_bare.jsonl
  ./Synthetic-JP-EN-Coding-Dataset-801k/train-00000-of-00013.parquet
  ./wizardlm8x22b-logical-math-coding-sft-ja/train-00000-of-00002.parquet

./mergekit/
  ./config.yml
  ./split1.yml

# ==================================================
# model weights
# ==================================================

./qwen3-150x52m-7.8b
  ./chat_template.jinja
  ./config.json
  ./mergekit_moe_config.yml
  ./model.safetensors.index.json
  ./model-00001-of-00002.safetensors
  ./model-00002-of-00002.safetensors
  ./tokenizer.json
  ./tokenizer_config.json

./qwen3-150x52m-7.8b-step1-distll
  ./chat_template.jinja
  ./config.json
  ./generation_config.json
  ./model.safetensors
  ./tokenizer.json
  ./tokenizer_config.json

./qwen3-150x52m-7.8b-step2-wikipedia
  ./chat_template.jinja
  ./config.json
  ./generation_config.json
  ./model.safetensors
  ./tokenizer.json
  ./tokenizer_config.json

./qwen3-150x52m-7.8b-step3-aozora
  ./chat_template.jinja
  ./config.json
  ./generation_config.json
  ./model.safetensors
  ./tokenizer.json
  ./tokenizer_config.json

./qwen3-150x52m-7.8b-step3-aozora-fix
  ./chat_template.jinja
  ./config.json
  ./generation_config.json
  ./model.safetensors
  ./tokenizer.json
  ./tokenizer_config.json

# ==================================================
# model gguf (can work upstream llama.cpp)
# ==================================================

Model Description

  • Developed by: [More Information Needed]
  • Funded by [optional]: [More Information Needed]
  • Shared by [optional]: [More Information Needed]
  • Model type: [More Information Needed]
  • Language(s) (NLP): [More Information Needed]
  • License: [More Information Needed]
  • Finetuned from model [optional]: [More Information Needed]

Model Sources [optional]

  • Repository: [More Information Needed]
  • Paper [optional]: [More Information Needed]
  • Demo [optional]: [More Information Needed]

Uses

Direct Use

[More Information Needed]

Downstream Use [optional]

[More Information Needed]

Out-of-Scope Use

[More Information Needed]

Bias, Risks, and Limitations

[More Information Needed]

Recommendations

Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.

How to Get Started with the Model

Use the code below to get started with the model.

[More Information Needed]

Training Details

Training Data

Dataset:

raw_text | books:

  • OmniAICreator/Japanese-Wikipedia-202506
  • globis-university/aozorabunko-clean

chat | reasoning:

  • DataPilot/Zero_SFT_Ja_v3.5_Reasoning

math | coding:

  • llm-jp/wizardlm8x22b-logical-math-coding-sft-ja
  • Aratako/Synthetic-JP-EN-Coding-Dataset-801k

Training Procedure

Preprocessing [optional]

None

Training Hyperparameters

  • Training regime: [More Information Needed]

  • qwen3-150x52m-step1-distill

    model (30000steps checkpoints)

    dataset

    • DataPilot/Zero_SFT_Ja_v3.5_Reasoning
    #### lora config
    
      task_type=TaskType.CAUSAL_LM,
      r=16,
      lora_alpha=32,
      target_modules=[
          "q_proj",
          "k_proj",
          "v_proj",
          "o_proj",
          "gate_proj",
          "up_proj",
          "down_proj",
          "gate_up_proj"
      ]
      lora_dropout=0,
      bias="none",
    
    #### transformer trainer config
      per_device_train_batch_size=1,
      gradient_accumulation_steps=4, 
      warmup_steps=5,
      max_steps=50000,
      learning_rate=2e-4,
      optim="adamw_8bit",             
      weight_decay=0.001,
      lr_scheduler_type="linear",
      seed=3407,
      fp16=False,                     
      bf16=True,                     
      remove_unused_columns=False
    
  • qwen3-150x52m-step2-wikipedia

    model (10000steps checkpoints)

    dataset

    • DataPilot/Zero_SFT_Ja_v3.5_Reasoning [:30000]
    • OmniAICreator/Japanese-Wikipedia-202506 [:30000]
    #### lora config
    
      task_type=TaskType.CAUSAL_LM,
      r=16,
      lora_alpha=32,
      target_modules=[
          "q_proj",
          "k_proj",
          "v_proj",
          "o_proj",
          "gate_proj",
          "up_proj",
          "down_proj",
          "gate_up_proj"
      ]
      lora_dropout=0,
      bias="none",
    
    #### transformer trainer config
      per_device_train_batch_size=1,
      gradient_accumulation_steps=4, 
      warmup_steps=5,
      max_steps=50000,
      learning_rate=2e-4,
      optim="adamw_8bit",             
      weight_decay=0.001,
      lr_scheduler_type="linear",
      seed=3407,
      fp16=False,                     
      bf16=True,                     
      remove_unused_columns=False
    
  • qwen3-150x52m-step3-aozora

    model (15000steps checkpoints)

    dataset

    • DataPilot/Zero_SFT_Ja_v3.5_Reasoning [:30000]
    • OmniAICreator/Japanese-Wikipedia-202506 [:30000]
    • globis-university/aozorabunko-clean [:15000]
    #### lora config
    
      task_type=TaskType.CAUSAL_LM,
      r=16,
      lora_alpha=32,
      target_modules=[
          "q_proj",
          "k_proj",
          "v_proj",
          "o_proj",
          "gate_proj",
          "up_proj",
          "down_proj",
          "gate_up_proj"
      ]
      lora_dropout=0,
      bias="none",
    
    #### transformer trainer config
      per_device_train_batch_size=1,
      gradient_accumulation_steps=4, 
      warmup_steps=5,
      max_steps=50000,
      learning_rate=2e-4,
      optim="adamw_8bit",             
      weight_decay=0.001,
      lr_scheduler_type="linear",
      seed=3407,
      fp16=False,                     
      bf16=True,                     
      remove_unused_columns=False
    
  • qwen3-150x52m-step3-aozora-fix

    model (45000steps checkpoints)

    dataset

    • DataPilot/Zero_SFT_Ja_v3.5_Reasoning [:30000]
    • OmniAICreator/Japanese-Wikipedia-202506 [:30000]
    • globis-university/aozorabunko-clean [:15000]
    #### lora config
      task_type=TaskType.CAUSAL_LM,
      r=16,
      lora_alpha=32,
      target_modules=[
          "q_proj",
          "k_proj",
          "v_proj",
          "o_proj",
          "gate_proj",
          "up_proj",
          "down_proj",
          "gate_up_proj"
      ]
      lora_dropout=0,
      bias="none",
    
    #### SFTtrainer SFTConfig
          max_length=max_8192,
          per_device_train_batch_size=1,
          gradient_accumulation_steps=1,
          warmup_steps=0,
          max_steps=60000,
          learning_rate=5e-4,
          logging_steps=5,
          optim="adamw_8bit",            
          weight_decay=0.0005,
          lr_scheduler_type="linear",
          seed=3407,
          report_to="none",
          bf16=True,                    
          remove_unused_columns=False,
          completion_only_loss=True
    
  • qwen3-150x52m-step4_math

    dataset

    • DataPilot/Zero_SFT_Ja_v3.5_Reasoning [:10000]
    • OmniAICreator/Japanese-Wikipedia-202506 [:10000]
    • globis-university/aozorabunko-clean [:10000]
    • llm-jp/wizardlm8x22b-logical-math-coding-sft-ja [:10000]
    • Aratako/Synthetic-JP-EN-Coding-Dataset-801k [:10000]
    #### lora config
      task_type=TaskType.CAUSAL_LM,
      r=16,
      lora_alpha=32,
      target_modules=[
          "q_proj",
          "k_proj",
          "v_proj",
          "o_proj",
          "gate_proj",
          "up_proj",
          "down_proj",
          "gate_up_proj"
      ]
      lora_dropout=0,
      bias="none",
    
    #### SFTtrainer SFTConfig
          max_length=8192,
          per_device_train_batch_size=1,
          gradient_accumulation_steps=1,
          warmup_steps=0,
          max_steps=60000,
          learning_rate=5e-4,
          logging_steps=5,
          optim="adamw_8bit",            
          weight_decay=0.0005,
          lr_scheduler_type="linear",
          seed=3407,
          report_to="none",
          bf16=True,                    
          remove_unused_columns=False,
          completion_only_loss=True
    
  • qwen3-150x52m-step4-math-fix

    
    

Speeds, Sizes, Times [optional]

[More Information Needed]

Evaluation

Testing Data, Factors & Metrics

Testing Data

[More Information Needed]

Factors

[More Information Needed]

Metrics

[More Information Needed]

Results

[More Information Needed]

Summary

Model Examination [optional]

[More Information Needed]

Environmental Impact

Carbon emissions can be estimated using the Machine Learning Impact calculator presented in Lacoste et al. (2019).

  • Hardware Type: [More Information Needed]
  • Hours used: [More Information Needed]
  • Cloud Provider: [More Information Needed]
  • Compute Region: [More Information Needed]
  • Carbon Emitted: [More Information Needed]

Technical Specifications [optional]

Model Architecture and Objective

Qwen3Moe

Compute Infrastructure

local pc

Hardware

Nvidia RTX 3060 * 1

Software

peft, trl

hou-rai (蓬莱) 2 Trainig codes | Other

logo image

Technical Specifications [optional]

Model Architecture and Objective

Qwen3Next

Compute Infrastructure

local pc

Hardware

Nvidia RTX 3060 * 1

Software

peft, trl


Citation [optional]

BibTeX:

[More Information Needed]

APA:

[More Information Needed]

Glossary [optional]

[More Information Needed]

More Information [optional]

[More Information Needed]

Model Card Authors [optional]

[More Information Needed]

Model Card Contact

[More Information Needed]

Downloads last month
238
GGUF
Model size
4B params
Architecture
qwen3moe
Hardware compatibility
Log In to add your hardware

8-bit

Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for WariHima/Hourai-a150m-7.8b-TrainingCodes-Demo

Finetuned
Qwen/Qwen3-0.6B
Quantized
(380)
this model

Datasets used to train WariHima/Hourai-a150m-7.8b-TrainingCodes-Demo

Paper for WariHima/Hourai-a150m-7.8b-TrainingCodes-Demo