Views
No views yet
pip install transformers datasets1import transformers
2from transformers import AutoConfig, AutoModel, AutoTokenizer
3
4tokenizer = AutoTokenizer.from_pretrained("drive/MyDrive/result/Qwen-DPO")
5
6message = [
7 {"role": "system", "content": "You are a helpful assistant chatbot."},
8 {"role": "user", "content": "What is a Large Language Model?"}
9]
10prompt = tokenizer.apply_chat_template(message, add_generation_prompt=True, tokenize=False)
11
12pipeline = transformers.pipeline(
13 "text-generation",
14 model="co-gy/Qwen-DPO",
15 tokenizer=tokenizer
16)
17
18sequences = pipeline(
19 prompt,
20 do_sample=True,
21 temperature=0.7,
22 top_p=0.9,
23 num_return_sequences=1,
24 max_length=200,
25)
26print(sequences[0]['generated_text'])