Unverified Commit 1fc0fa46 authored by kumapo's avatar kumapo Committed by GitHub
Browse files

Make sure to raise NotImplementedError with correct method name (#15253)

parent f00f22a3
......@@ -600,7 +600,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
"""
Initialize the weights. This method should be overridden by derived class.
"""
raise NotImplementedError(f"Make sure `_init_weigths` is implemented for {self.__class__}")
raise NotImplementedError(f"Make sure `_init_weights` is implemented for {self.__class__}")
def tie_weights(self):
"""
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment