Update modeling_steerling.py

modeling_steerling.py

@@ -1,5 +1,5 @@
 from __future__ import annotations
-# Auto-generated by scripts/build_hf_files_v3.py — do not edit manually.
+# Auto-generated by scripts/build_hf_files.py — do not edit manually.
 
 import torch
 import torch.nn as nn
@@ -1562,6 +1562,9 @@ class SteerlingForCausalLM(PreTrainedModel):
     def _init_weights(self, module):
         pass
 
+    _forward_with_injection = InterpretableCausalDiffusionLM._forward_with_injection
+    _compute_unknown_topk = InterpretableCausalDiffusionLM._compute_unknown_topk
+
     def _tie_weights(self):
         if self.config.weight_sharing:
             self.transformer.lm_head.weight = self.transformer.tok_emb.weight