Views
No views yet
1import torch
2from transformers import AutoTokenizer, AutoModelForCausalLM
3from peft import PeftModel
4
5base_model = AutoModelForCausalLM.from_pretrained(
6 "ilsp/Llama-Krikri-8B-Base",
7 torch_dtype=torch.bfloat16,
8 device_map="auto"
9)
10tokenizer = AutoTokenizer.from_pretrained("ilsp/Llama-Krikri-8B-Base")
11model = PeftModel.from_pretrained(base_model, "Stergios/krikri-8b-base-lora")
12
13prompt = "Γράψε στην ποντιακή διάλεκτο: Καλημέρα, πώς είσαι;"
14inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
15with torch.no_grad():
16 output = model.generate(**inputs, max_new_tokens=120, temperature=0.7)
17print(tokenizer.decode(output[0], skip_special_tokens=True))<po>, <cr>, <no>, <cy>) mapped to natural Greek instructions using convert_to_natural_prompts_dialects_only.py