Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2from peft import PeftModel
3
4# Load tokenizer
5tokenizer = AutoTokenizer.from_pretrained("ErkaMarka/mongolian-mistral-7b-chatbot")
6
7# Load base model
8base_model = AutoModelForCausalLM.from_pretrained(
9 "mistralai/Mistral-7B-Instruct-v0.2",
10 torch_dtype=torch.float16,
11 device_map="auto"
12)
13
14# Resize embeddings for new tokens
15base_model.resize_token_embeddings(len(tokenizer))
16
17# Load LoRA adapter
18model = PeftModel.from_pretrained(base_model, "ErkaMarka/mongolian-mistral-7b-chatbot")
19
20# Generate
21messages = [{"role": "user", "content": "Монгол улсын нийслэл хот юу вэ?"}]
22text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
23inputs = tokenizer(text, return_tensors="pt").to(model.device)
24
25outputs = model.generate(**inputs, max_new_tokens=150)
26print(tokenizer.decode(outputs[0], skip_special_tokens=True))@misc{mongolian_mistral_7b_chatbot},
author = {Your Name},
title = {mongolian-mistral-7b-chatbot},
year = {2024},
publisher = {Hugging Face},
url = {https://huggingface.co/ErkaMarka/mongolian-mistral-7b-chatbot}
}