Update modeling_super_linear.py
Browse files- modeling_super_linear.py +0 -8
modeling_super_linear.py
CHANGED
|
@@ -541,11 +541,3 @@ class SuperLinearForCausalLM(PreTrainedModel, GenerationMixin):
|
|
| 541 |
def _reorder_cache(self, past, beam_idx, **kwargs):
|
| 542 |
return past # backbone keeps no KV cache
|
| 543 |
|
| 544 |
-
"-------------------------------------------------------------------------------------------------------------------"
|
| 545 |
-
# 3) --------------------------------------------------------------------------
|
| 546 |
-
# REGISTRATION (one‑liner you run **once** before .from_pretrained)
|
| 547 |
-
# -----------------------------------------------------------------------------
|
| 548 |
-
|
| 549 |
-
|
| 550 |
-
AutoConfig.register(SuperLinearConfig.model_type, SuperLinearConfig)
|
| 551 |
-
AutoModelForCausalLM.register(SuperLinearConfig, SuperLinearForCausalLM)
|
|
|
|
| 541 |
def _reorder_cache(self, past, beam_idx, **kwargs):
|
| 542 |
return past # backbone keeps no KV cache
|
| 543 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|