Views
No views yet
SFTTrainer
with Unsloth's 4-bit fast patching.| Param | Value |
|---|---|
| Base model | unsloth/tinyllama-bnb-4bit |
| Trainer | trl.SFTTrainer (Unsloth patched) |
| Data | Raw PDF paragraphs (9 records, packing=True) |
| Max steps | 30 |
| Learning rate | 2e-4 |
| LoRA r | 16 |
| LoRA alpha | 32 |
| Peak VRAM | 0.98 GB |
| Train time | 161s |
1from unsloth import FastLanguageModel
2
3model, tokenizer = FastLanguageModel.from_pretrained(
4 model_name="unsloth/tinyllama-bnb-4bit",
5 max_seq_length=512,
6 load_in_4bit=True,
7)
8
9from peft import PeftModel
10model = PeftModel.from_pretrained(model, "ThakrePranjal/pharma-tinyllama-unsloth-stage1-lora")unsloth/tinyllama-bnb-4bit
└── Stage 1 SFT (THIS ADAPTER) → merged → [ThakrePranjal/pharma-tinyllama-unsloth-stage1-merged]
└── Stage 2 SFT → merged → [ThakrePranjal/pharma-tinyllama-unsloth-stage2-merged]
└── Stage 3 DPO → merged → [ThakrePranjal/pharma-tinyllama-unsloth-final]