Views
No views yet
unsloth/llama-3-8b-bnb-4bitSFTTrainerDo you sell pizza?
1from transformers import AutoModelForCausalLM, AutoTokenizer
2import torch
3
4base_model = "unsloth/llama-3-8b-bnb-4bit"
5adapter = "your-username/sarcastic-bakery-lora"
6
7tokenizer = AutoTokenizer.from_pretrained(base_model)
8model = AutoModelForCausalLM.from_pretrained(
9 base_model,
10 load_in_4bit=True,
11 device_map="auto"
12)
13
14model.load_adapter(adapter)
15
16prompt = "Do you sell pizza?"
17inputs = tokenizer(prompt, return_tensors="pt")
18outputs = model.generate(**inputs, max_new_tokens=100)
19
20print(tokenizer.decode(outputs[0], skip_special_tokens=True))