Views
No views yet
1- **Developed by:** Fangyuan Yu
2- **Funded by Hardeep:** [Temus]
3- **Language(s) (NLP):** Thai, English
4- **License:** MIT
5- **Finetuned from model [optional]:** [Typhoon-7B]1from peft import PeftModel, PeftConfig
2from transformers import AutoModelForCausalLM
3
4config = PeftConfig.from_pretrained("Ksgk-fy/thaimus-7B")
5model = AutoModelForCausalLM.from_pretrained("scb10x/typhoon-7b")
6model = PeftModel.from_pretrained(model, "Ksgk-fy/thaimus-7B")
7pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
8
9messages = [{"role": "user", "content": "สวัสดีครับ/ค่ะ คุณเป็นอย่างไรบ้างครับ/ค่ะ?"}]
10prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
11outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
12print(outputs[0]["generated_text"])