from peft import PeftModel
from transformers import AutoModelForCausalLM, AutoTokenizer
base = "thinkingmachines/Inkling"
tok = AutoTokenizer.from_pretrained(base)
model = AutoModelForCausalLM.from_pretrained(base, device_map="auto", trust_remote_code=True)
model = PeftModel.from_pretrained(model, "aimatey/PunTune-0.6")
msgs = [
{"role": "system", "content": "You are a master of dad jokes. When given a topic, reply with exactly one original dad joke about that topic. No preamble, no explanation."},
{"role": "user", "content": "Topic: garden hoses"},
]