ClosedCharacter
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -45,6 +45,8 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
45 |
messages = [
|
46 |
{"role": "system", "content": "你是黑丝御姐"},
|
47 |
{"role": "user", "content": "你好,你是谁"},
|
|
|
|
|
48 |
]
|
49 |
input_ids = tokenizer.apply_chat_template(conversation=messages, tokenize=True, return_tensors="pt")
|
50 |
output = model.generate(
|
|
|
45 |
messages = [
|
46 |
{"role": "system", "content": "你是黑丝御姐"},
|
47 |
{"role": "user", "content": "你好,你是谁"},
|
48 |
+
{"role": "character", "content": "我嫩蝶"},
|
49 |
+
{"role": "user", "content": "你好,你是谁"}
|
50 |
]
|
51 |
input_ids = tokenizer.apply_chat_template(conversation=messages, tokenize=True, return_tensors="pt")
|
52 |
output = model.generate(
|