okestro-ai-lab
/

FastSLM

Audio-Text-to-Text

feature-extraction

text-generation

Model card Files Files and versions

speechisalluneed commited on Aug 5

Commit

e08aed4

·

verified ·

1 Parent(s): b441b7e

Update modeling_fastslm.py

Files changed (1) hide show

modeling_fastslm.py +1 -1

modeling_fastslm.py CHANGED Viewed

@@ -289,7 +289,7 @@ class SpeechEncoder(nn.Module):
 class FastSLMPreTrainedModel(PreTrainedModel):
     config_class = FastSLMConfig
     base_model_prefix = "fastslm"
     def _init_weights(self, module):
         if isinstance(module, nn.Linear):
             nn.init.normal_(module.weight, std=0.02)

 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)