ClosedCharacter commited on
Commit
f5d5579
·
verified ·
1 Parent(s): 7a6a8bb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
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(