>>103297151
>if you do make a PEFT version
Well, I've already made the PEFT version, it's just not released yet. But I'll probably release it at the end of today or tomorrow, I just need to take a break. But the PEFT version is already integrated into HF peft and be accessed just how you'd access any PEFT model.
from peft import SMTConfig, SMTModel, get_peft_model
config = SMTConfig(
peft_type="SMT",
task_type="CAUSAL_LM",
target_modules=["q_proj", "v_proj", "k_proj"],
sparsity_ratio=0.05,
block_size=256,
selection_method="GW",
dataloader=dataloader,
)
model = get_peft_model(model, config)