Update modeling_fastslm.py

modeling_fastslm.py

@@ -289,7 +289,7 @@ class SpeechEncoder(nn.Module):
 class FastSLMPreTrainedModel(PreTrainedModel):
     config_class = FastSLMConfig
     base_model_prefix = "fastslm"
-
+    self.post_init()
     def _init_weights(self, module):
         if isinstance(module, nn.Linear):
             nn.init.normal_(module.weight, std=0.02)
@@ -326,6 +326,8 @@ class FastSLMForConditionalGeneration(FastSLMPreTrainedModel, GenerationMixin):
         language_token = [f"<|{lang.upper()}|>" for lang in LANGUAGES]
         special_tokens = audio_token + language_token + task_token
         self.tokenizer.add_special_tokens({"additional_special_tokens": special_tokens})
+
+        self.post_init()
         
     def get_input_embeddings(self) -> nn.Module:
         """Returns the input embedding layer of the LLM."""