spadeMIA/GoodWiki_Corpus_1024_2040
Viewer • Updated • 11.7k • 218
How to use spadeMIA/pythia-2.8b-goodwiki-lora-r64 with PEFT:
from peft import PeftModel
from transformers import AutoModelForCausalLM
base_model = AutoModelForCausalLM.from_pretrained("EleutherAI/pythia-2.8b")
model = PeftModel.from_pretrained(base_model, "spadeMIA/pythia-2.8b-goodwiki-lora-r64")LoRA-fine-tuned checkpoint produced for membership inference research at the SPADE Lab, Koç University.
EleutherAI/pythia-2.8bspadeMIA/GoodWiki_Corpus_1024_2040train (10,000 rows, all membership-positive)| Hyperparameter | Value |
|---|---|
| Rank (r) | 64 |
| Alpha | 128 |
| Dropout | 0.05 |
| Target modules | query_key_value, dense, dense_h_to_4h, dense_4h_to_h |
| Hyperparameter | Value |
|---|---|
| Epochs | 2 |
| Learning rate | 0.0002 (constant) |
| LR scheduler | constant_with_warmup |
| Warmup ratio | 0.01 |
| Weight decay | 0.0 |
| Max grad norm | 1.0 |
| Per-device batch size | 1 |
| Gradient accumulation steps | 32 |
| Effective batch size | 32 |
| Precision | bf16 |
| TF32 matmul | True |
| Gradient checkpointing | True |
| Optimizer | adamw_torch |
| Seed | 42 |
| Checkpointing | none during training; only the final adapter is saved |
test split was not loaded or trained on.train[:50].Generated automatically by run_lora.py on 2026-08-12.
Base model
EleutherAI/pythia-2.8b