Views
No views yet
| Folder | Dataset(s) Used | Description |
|---|---|---|
llama-3.2-3B-sft | Alpaca | Fine-tuned only on the original Alpaca dataset |
llama-3.2-3B-sft-dolly | Alpaca + Dolly | Fine-tuned on Databricks' Dolly dataset |
llama-3.2-3B-sft-FLAN | Alpaca + Dolly + FLAN | Fine-tuned on FLAN and Alpaca mixed |
sft_a_d | Alpaca + Dolly | Combined dataset fine-tuning (Alpaca + Dolly) |
sft_a_d1 | Alpaca(cleaned) + Dolly | Combined dataset fine-tuning (Alpaca + Dolly) |
peft)1from peft import PeftModel
2from transformers import AutoTokenizer, AutoModelForCausalLM
3
4# Load base model
5base_model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-3B")
6tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-3B")
7
8# Load adapter (choose one)
9model = PeftModel.from_pretrained(base_model, "gg-cse476/gg/sft_a_d")
10
11# Inference
12prompt = "Explain how a rocket works in simple terms."
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))