gemma-wang-zh

Merged Wang Yangming style adapter on top of google/gemma-4-E4B-it.

This repository contains the merged full Transformers model, not only the PEFT adapter. The adapter was trained with Chinese Wang Yangming/persona SFT data at max length 1536, LoRA r=8 alpha=16.

Quick use

import torch
from transformers import AutoModelForCausalLM, AutoProcessor

repo = "xihajun/gemma-wang-zh"
processor = AutoProcessor.from_pretrained(repo)
model = AutoModelForCausalLM.from_pretrained(repo, torch_dtype=torch.bfloat16, device_map="auto")

messages = [
    {"role": "system", "content": "ไฝ ๅฐ†ไปฅ็Ž‹้˜ณๆ˜Ž็š„ๅฃๅปๅ’Œๅฟƒๅญฆๆ€ๆƒณๅ›ž็ญ”๏ผšๅผบ่ฐƒ่‡ด่‰ฏ็Ÿฅใ€็Ÿฅ่กŒๅˆไธ€ใ€ไบ‹ไธŠ็ฃจ็‚ผใ€‚ไธ่ฆ่งฃ้‡Š้ฃŽๆ ผ๏ผŒ็›ดๆŽฅๅ›ž็ญ”ใ€‚"},
    {"role": "user", "content": "ๅฆ‚ๆžœๆˆ‘ๆƒณๅพˆๅคšไฝ†่กŒๅŠจๅพˆๅฐ‘๏ผŒๅบ”่ฏฅๆ€ŽไนˆๅŠž๏ผŸ"},
]
prompt = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True, enable_thinking=False)
inputs = processor(text=[prompt], return_tensors="pt").to(model.device)
out = model.generate(**inputs, max_new_tokens=128, do_sample=False)
print(processor.decode(out[0, inputs["input_ids"].shape[-1]:], skip_special_tokens=True))

Notes

This is a merged Hugging Face/Transformers model. It is not a .litertlm mobile package yet.

Downloads last month
16
Safetensors
Model size
8B params
Tensor type
BF16
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support

Model tree for xihajun/gemma-wang-zh

Adapter
(433)
this model