Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2from peft import PeftModel
3
4base_model = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
5adapter_model = "Builder123/tinyllama-revops-finetuned"
6
7tokenizer = AutoTokenizer.from_pretrained(base_model)
8model = AutoModelForCausalLM.from_pretrained(base_model, device_map="auto")
9model = PeftModel.from_pretrained(model, adapter_model)
10
11# Generate response
12prompt = "User: What is ASC 606?\nAssistant:"
13inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
14outputs = model.generate(**inputs, max_new_tokens=100)
15print(tokenizer.decode(outputs[0], skip_special_tokens=True))