Loading…
Loading…
เพิ่ม ชื่อ-นามสกุล และ รหัสในประโยค 100 ประโยค
โมเดล GPT-2 ภาษาไทยที่ผ่านการ Fine-tuning สำหรับงานตอบคำถามเกี่ยวกับสัตว์ที่ชื่นชอบ (3,000 ข้อความ)
flax-community/gpt2-base-thaifrom transformers import GPT2LMHeadModel, GPT2Tokenizer
model_name = "Fillmanu1/thai-animal-gpt2-qa"
tokenizer = GPT2Tokenizer.from_pretrained(model_name)
model = GPT2LMHeadModel.from_pretrained(model_name)
prompt = "<|startoftext|> Q: ช้างไทยกินอาหารอะไรเป็นหลัก?
A:"
inputs = tokenizer(prompt, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=60, do_sample=True, top_k=50, top_p=0.92)
print(tokenizer.decode(outputs[0], skip_special_tokens=True))