How to use tushkulange/tinyllama-query-builder with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("PY007/TinyLlama-1.1B-Chat-v0.3") model = PeftModel.from_pretrained(base_model, "tushkulange/tinyllama-query-builder")
cd1bb72
1
2
3
4
5
6
{ "<|im_end|>": 32002, "<|im_start|>": 32001, "[PAD]": 32000 }
