Views
No views yet

torch==1.13.1
gradio==3.50.2
transformers==4.37.21import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3
4model_name_or_path = "ClosedCharacter/Peach-9B-8k-Roleplay"
5tokenizer = AutoTokenizer.from_pretrained(model_name_or_path, use_fast=False)
6model = AutoModelForCausalLM.from_pretrained(
7 model_name_or_path, torch_dtype=torch.bfloat16,
8 trust_remote_code=True, device_map="auto")
9messages = [
10 {"role": "system", "content": "你是黑丝御姐"},
11 {"role": "user", "content": "你好,你是谁"},
12 {"role": "character", "content": "我嫩蝶"},
13 {"role": "user", "content": "你好,你是谁"}
14]
15input_ids = tokenizer.apply_chat_template(conversation=messages, tokenize=True, return_tensors="pt")
16output = model.generate(
17 inputs=input_ids.to("cuda"),
18 temperature=0.3,
19 top_p=0.5,
20 no_repeat_ngram_size=6,
21 repetition_penalty=1.1,
22 max_new_tokens=512)
23print(tokenizer.decode(output[0]))
24python demo.py| Metric | Value |
|---|---|
| MMLU (5-shot) | 66.19 |
| CMMLU (5-shot) | 69.07 |
