marcelbinz/Psych-101
Viewer • Updated • 60.1k • 689 • 122
How to use socius/Llama-Centaur-8B-LoRA-r4 with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("unsloth/Llama-3.1-8B")
model = PeftModel.from_pretrained(base_model, "socius/Llama-Centaur-8B-LoRA-r4")LoRA adapter for Llama-Centaur-8B, 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.1-8B |
| LoRA rank | 4 (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.1-8B, or evaluate with the project's
eval_model.py --backend unsloth.