rp-glm-4.7-flash-lora

RP-GLM-4.7-Flash LoRA

基于 GLM-4.7-Flash 的角色扮演 LoRA 适配器。

模型描述

这是一个 LoRA (Low-Rank Adaptation) 适配器，专门针对角色扮演场景进行了微调。

特点

• 🎭 角色扮演对话能力
• 📝 创意写作与叙事
• 💬 多轮对话连贯性
• 🌏 中英双语支持
• 📦 轻量级，易于部署

训练配置

使用方法

方法 1: 使用 PEFT 加载

from transformers import AutoModelForCausalLM, AutoTokenizer
from peft import PeftModel

base_model_name = "unsloth/GLM-4.7-Flash"
lora_name = "taozi555/rp-glm-4.7-flash-lora"

# 加载基础模型
tokenizer = AutoTokenizer.from_pretrained(base_model_name, trust_remote_code=True)
model = AutoModelForCausalLM.from_pretrained(
    base_model_name,
    torch_dtype="auto",
    device_map="auto",
    trust_remote_code=True,
)

# 加载 LoRA
model = PeftModel.from_pretrained(model, lora_name)

# 推理
messages = [
    {"role": "system", "content": "你是一位来自中世纪的强大魔法师，名叫梅林。"},
    {"role": "user", "content": "伟大的梅林，请问您能教我魔法吗？"}
]

text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
inputs = tokenizer(text, return_tensors="pt").to(model.device)

outputs = model.generate(**inputs, max_new_tokens=512, temperature=0.7, top_p=0.9)
response = tokenizer.decode(outputs[0][inputs["input_ids"].shape[1]:], skip_special_tokens=True)
print(response)

方法 2: 合并后使用

from peft import PeftModel

# 加载后合并
model = PeftModel.from_pretrained(base_model, lora_name)
model = model.merge_and_unload()  # 合并权重

# 现在可以像普通模型一样使用

相关链接

• 🔗 合并后的完整模型: taozi555/rp-glm-4.7-flash
• 🔗 基础模型: unsloth/GLM-4.7-Flash

注意事项

• 本 LoRA 专门针对角色扮演场景优化
• 推荐使用 temperature=0.7, top_p=0.9 进行采样
• 需要与基础模型 unsloth/GLM-4.7-Flash 配合使用

License

Apache 2.0