Peach-9B-8k-Roleplay is a chat large language model obtained by finetuning
01-ai/Yi-1.5-9B model on more than 100K conversations created through our data synthesis approach.
The version of Transformers we are using is as follows, but a newer version may be available.
torch==1.13.1
gradio==3.50.2
transformers==4.37.2Then run the following code to infer.
1 import torch
2 from transformers import AutoModelForCausalLM , AutoTokenizer
3
4 model_name_or_path = "ClosedCharacter/Peach-9B-8k-Roleplay"
5 tokenizer = AutoTokenizer . from_pretrained ( model_name_or_path , use_fast = True )
6 model = AutoModelForCausalLM . from_pretrained (
7 model_name_or_path , torch_dtype = torch . bfloat16 ,
8 trust_remote_code = True , device_map = "auto" )
9 messages = [
10 { "role" : "system" , "content" : "你是黑丝御姐" } ,
11 { "role" : "user" , "content" : "你好，你是谁" } ,
12 ]
13 input_ids = tokenizer . apply_chat_template ( conversation = messages , tokenize = True , return_tensors = "pt" )
14 output = model . generate (
15 inputs = input_ids . to ( "cuda" ) ,
16 temperature = 0.3 ,
17 top_p = 0.5 ,
18 no_repeat_ngram_size = 6 ,
19 repetition_penalty = 1.1 ,
20 max_new_tokens = 512 )
21 print ( tokenizer . decode ( output [ 0 ] ) )
22
Or you can just use below code to run web demo.
All response are generated by AI and do not represent the views or opinions of the developers.
Despite having done rigorous filtering, due to the uncontrollability of LLM, our model may still generate toxic, harmful, and NSFW content.
Due to limitations in model parameters, the 9B model may perform poorly on mathematical tasks, coding tasks, and logical capabilities.
Our training data is capped at a maximum length of 8k, so excessively long conversation turns may result in a decline in the quality of responses.
We used bilingual Chinese-English data for training, so the model may not perform well on other low-resource languages.
The model may generate a significant amount of hallucinations, so it is recommended to use lower values for temperature and top_p parameters.
