初始化项目,由ModelHub XC社区提供模型

Model: saadxsalman/ss-medical-moe
Source: Original Platform
This commit is contained in:
ModelHub XC
2026-07-09 23:06:25 +08:00
commit c66cd5146c
13 changed files with 265 additions and 0 deletions

35
moe_config.json Normal file
View File

@@ -0,0 +1,35 @@
{
"base_model": "meta-llama/Llama-3.2-1B",
"hidden_size": 2048,
"ffn_hidden": 4096,
"lora_r": 16,
"lora_alpha": 32,
"lora_dropout": 0.05,
"lora_targets": [
"q_proj",
"v_proj",
"k_proj",
"o_proj"
],
"num_experts": 4,
"top_k": 2,
"dataset_name": "Malikeh1375/medical-question-answering-datasets",
"dataset_subset": "all-processed",
"num_samples": 15000,
"max_length": 256,
"batch_size": 2,
"grad_accum": 16,
"warmup_ratio": 0.1,
"val_split": 0.1,
"save_path": "/kaggle/working/llama_moe_medical",
"model_type": "llama-3.2-1b-lora-moe",
"lora_merged": true,
"hf_username": "saadxsalman",
"repo_id": "saadxsalman/ss-medical-moe",
"domain_experts": {
"expert_1": "Symptoms & Diagnosis",
"expert_2": "Drug Interactions",
"expert_3": "Anatomy & Physiology",
"expert_4": "Treatment & Procedures"
}
}