Views
No views yet
trl ร่วมกับ QLoRA บนฐาน mistralai/Mistral-7B-Instruct-v0.31from transformers import pipeline, AutoTokenizer, AutoModelForCausalLM
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained("UpMath/Thai-HomeworkGen-7.2B-v1")
5model = AutoModelForCausalLM.from_pretrained("UpMath/Thai-HomeworkGen-7.2B-v1",device_map="auto")
6
7generator = pipeline(
8 "text-generation",
9 model=model,
10 tokenizer=tokenizer,
11 device_map="auto",
12 max_new_tokens=2048,
13 do_sample=True,
14 temperature=0.7,
15 pad_token_id=tokenizer.eos_token_id
16)
17
18prompt = """
19 ระดับ Bloom: "วิเคราะห์" # หรือ ["วิเคราะห์", "จดจำ"]
20 ระดับชั้น: 'มัธยมศึกษาปีที่ 4'
21 โปรดสร้างโจทย์ พร้อมวิธีทำและคำตอบ:
22"""
23
24result = generator(prompt)[0]['generated_text']
25print(result)mistralai/Mistral-7B-Instruct-v0.3trl (SFTTrainer)peft)