Yida-Model-MM

Merged full-weight checkpoint of a Qwen3-14B LoRA supervised fine-tune. Hub files are merged bfloat16 weights, not a PEFT adapter.

This model is for research and engineering evaluation. It is not a medical device and must not be used as the sole basis for clinical decisions.

Model details

Item Value
Developed by rwang220
Model type Causal LM (Qwen3ForCausalLM)
Parameters 14.77B (14,768,307,200)
Precision bfloat16
Architecture 40 layers, hidden size 5120, GQA 40/8, intermediate size 17408, vocab 151936
Context max_position_embeddings 40,960. SFT max length 8,192
License Apache 2.0, inherited from Qwen3-14B

Training

Fine-tuned with ms-swift LoRA, then merged into full weights.

Item Value
Immediate base Public Qwen/Qwen3-14B.
Run mm_sft_5_29/v0-20260529-080809
LoRA rank 16, alpha 32, dropout 0.05, targets q/k/v/o/gate/up/down_proj
Optim AdamW, lr 1e-4, cosine, warmup ratio 0.05
Schedule 3 epochs, 72,348 steps, max length 8192
Batch per-device 1, gradient accumulation 8
Best eval loss 0.6525 at step 71,500
Final eval loss 0.6525
Final eval token acc 0.7803

Data

An internal, deduplicated Chinese medical instruction mix. Categories were capped; this is not the full uncapped set used by the DS variant.

How to use

from transformers import AutoModelForCausalLM, AutoTokenizer

model_id = "rwang220/Yida-Model-MM"
tokenizer = AutoTokenizer.from_pretrained(model_id)
model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype="auto", device_map="auto")
messages = [{"role": "user", "content": "请用通俗语言解释高血压的常见注意事项。"}]
text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True, enable_thinking=True)
inputs = tokenizer([text], return_tensors="pt").to(model.device)
output = model.generate(**inputs, max_new_tokens=1024, temperature=0.6, top_p=0.95, top_k=20)
print(tokenizer.decode(output[0][inputs.input_ids.shape[1]:], skip_special_tokens=True))
vllm serve rwang220/Yida-Model-MM --reasoning-parser qwen3

Limitations

  • Can hallucinate guidelines, doses, diagnoses, and citations.
  • Training data is internal and not released with this repository.
  • Reported numbers are training eval loss and token accuracy only. No clinical benchmark is claimed.
Downloads last month
301
Safetensors
Model size
15B params
Tensor type
BF16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support