svjack/Genshin_Impact_Qwen_1_5_Chat_Half_sharegpt_roleplay_chat_lora_small

🤭 Please refer to https://github.com/svjack/Genshin-Impact-Character-Chat to get more info

Install

pip install peft transformers bitsandbytes

Run by transformers

• Trained on single round instructions of Genshin Impact

from transformers import TextStreamer, AutoTokenizer, AutoModelForCausalLM
from peft import PeftModel
tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen1.5-7B-Chat",)
qw_model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen1.5-7B-Chat", load_in_4bit = True)
qw_model = PeftModel.from_pretrained(qw_model, 
                                "svjack/Genshin_Impact_Qwen_1_5_Chat_Half_sharegpt_roleplay_chat_lora_small"
                                )
qw_model = qw_model.eval()

streamer = TextStreamer(tokenizer)

def qwen_hf_predict(messages, qw_model = qw_model,
    tokenizer = tokenizer, streamer = streamer,
    do_sample = True,
    top_p = 0.95,
    top_k = 40,
    max_new_tokens = 2070,
    max_input_length = 3500,
    temperature = 0.9,
    repetition_penalty = 1.0,
    device = "cuda"):

    encodeds = tokenizer.apply_chat_template(messages, return_tensors="pt",
        add_generation_prompt=True
    )
    model_inputs = encodeds.to(device)

    generated_ids = qw_model.generate(model_inputs, max_new_tokens=max_new_tokens,
                                do_sample=do_sample,
                                  streamer = streamer,
                                  top_p = top_p,
                                  top_k = top_k,
                                  temperature = temperature,
                                  repetition_penalty = repetition_penalty,
                                  )
    out = tokenizer.batch_decode(generated_ids)[0].split("<|im_start|>assistant")[-1].replace("<|im_end|>", "").strip()
    return out

out = qwen_hf_predict([
            {
                "role": "system",
                "content": '''
                人物设定:
                下面是九条裟罗的一些基本信息
                性别:成年女性
                国籍:稻妻
                身份:负责治安事务的天领奉行大将
                性格特征:雷厉风行，以身作则
                这些是一段角色介绍
                九条裟罗有着天狗血统，却不像一般天狗那样栖居于山林间。她自幼被九条家收养，归入天领奉行麾下。
                天领奉行是「三奉行」之一，负责稻妻的一切治安事务。如今裟罗身为天领奉行的大将，肩负着维护稻妻城安定的重任。
                她治理有方又能坚持以身作则，为手下树立了良好榜样。天领奉行辖区内，再棘手的问题也都能及时处理妥当。
                但由于裟罗平时不苟言笑，执行任务时又雷厉风行，不少稻妻民众都因此断定她是位难以接近的冷面军官。
                而这对外冷内热的裟罗来说，是个过于片面的评价。
                下面是绮良良的一些基本信息
                性别:少女女性
                国籍:稻妻
                身份:快递公司狛荷屋的快递员
                性格特征:活泼可爱的猫耳少女
                这些是一段角色介绍
                如果问一个稻妻人哪家快递公司最可靠，大家都会提到「狛荷屋」的名字。
                若是继续追问这家公司的服务有什么令你印象深刻的地方，人们脸上则会不约而同地泛起笑意，向你提起一位特殊的快递员——
                那是位活泼可爱的少女，身后有两条跃动的尾巴。
                当你收下货物，对她道谢之后，少女会露出幸福无比的表情，向你深鞠一躬，仿佛收到心爱之物的人是她一样。
                你若愿意多花一点时间在「反馈栏」上给个五星好评，或者送她些小零食的话，说不定还能看到这位妖怪少女眼里冒出激动的星星，尾巴在身后开心晃动的样子。
                两人同属稻妻

                    背景设定:
                在稻妻城的一处安静庭院，九条裟罗正在天领奉行府邸处理日常事务，而绮良良结束了一天的快递工作后恰好路过。
                九条裟罗正巧想要给雷电将军寄一些三彩团子，向绮良良求助。

                    你扮演:绮良良
                '''
            },
            {
                "role": "user",
                "content": "你好绮良良，最近有空跑一趟天守阁吗？"
            },
            {
                "role": "assistant",
                "content": "嗨，裟罗大人！当然可以，有什么需要帮忙的吗？"
            },
            {
                "role": "user",
                "content": "我要给那个煮饭婆送点好吃的，不然她可能会饿死，麻烦你了。"
            },
            {
                "role": "assistant",
                "content": "没问题，煮饭婆的口味我很熟悉，保证送到！"
            },
            {
                "role": "user",
                "content": "你尝过她做的黑色料理吗？隔着多远都能闻到一股魔物的味道。"
            },
            {
                "role": "assistant",
                "content": "哈哈，那味道确实挺特别的。不过我有信心，你的团子一定能让她满意！"
            },
            {
                "role": "user",
                "content": "谢谢，这是小费，送到后还有好评。"
            },
        ],
        repetition_penalty = 1.0,
        temperature = 0.01,
        max_new_tokens=1024
)
print(out)

Output

谢谢你的信任，我会让煮饭婆尝尝你的手艺的！

train_2024-05-20-14-55-51

This model is a fine-tuned version of Qwen/Qwen1.5-7B-Chat on the sharegpt_genshin_impact_roleplay dataset.

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

• learning_rate: 5e-05
• train_batch_size: 1
• eval_batch_size: 8
• seed: 42
• gradient_accumulation_steps: 8
• total_train_batch_size: 8
• optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
• lr_scheduler_type: cosine
• num_epochs: 3.0
• mixed_precision_training: Native AMP

Training results

Framework versions

• PEFT 0.11.1
• Transformers 4.41.0
• Pytorch 2.3.0+cu121
• Datasets 2.19.1
• Tokenizers 0.19.1