Update modeling_steerling.py

modeling_steerling.py

@@ -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