orionweller
/

test-flex-gpt

Model card Files Files and versions Community

oweller2 commited on about 1 month ago

Commit

47fa45e

•

1 Parent(s): f83d8e4

working

Files changed (1) hide show

modeling_flexbert.py +1 -1

modeling_flexbert.py CHANGED Viewed

@@ -1536,7 +1536,7 @@ class FlexBertForCasualLM(FlexBertPreTrainedModel):
         # Initialize weights and apply final processing
         self._init_weights(reset_params=False)
-[]    def _init_weights(self, module: Optional[nn.Module] = None, reset_params: Optional[bool] = None):
         # Handle the XOR condition
         assert (module is None) != (reset_params is None), "arg module xor reset_params must be specified"

         # Initialize weights and apply final processing
         self._init_weights(reset_params=False)
+    def _init_weights(self, module: Optional[nn.Module] = None, reset_params: Optional[bool] = None):
         # Handle the XOR condition
         assert (module is None) != (reset_params is None), "arg module xor reset_params must be specified"