marcelbinz/Psych-101
Viewer • Updated • 60.1k • 675 • 122
How to use socius/Llama-Centaur-3B-LoRA-r16 with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("unsloth/Llama-3.2-3B")
model = PeftModel.from_pretrained(base_model, "socius/Llama-Centaur-3B-LoRA-r16")LoRA adapter for Llama-Centaur-3B, fine-tuned on the full Psych-101 as part of the LoRA-rank sweep and dataset-size ablation for Small Foundation Models of Human Cognition and Behaviour.
| field | value |
|---|---|
| base model | unsloth/Llama-3.2-3B |
| LoRA rank | 16 (alpha = rank, rsLoRA) |
| data fraction | 100% of Psych-101 |
| training | 1 epoch, completion-only loss, seed 3407 |
Load with PEFT on top of unsloth/Llama-3.2-3B, or evaluate with the project's
eval_model.py --backend unsloth.