Skip to content
Prev Previous commit
Next Next commit
Apply suggestion from @kashif
  • Loading branch information
kashif authored Dec 5, 2025
commit b64d9db0b02e7412cc86b34f736955253c4f37b8
3 changes: 1 addition & 2 deletions src/chronos/chronos2/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -272,8 +272,7 @@ def __init__(self, config: Chronos2CoreConfig):
self.post_init()

def _init_weights(self, module):
"""Initialize the weights.
"""
"""Initialize the weights"""
factor = self.config.initializer_factor
if isinstance(module, Chronos2LayerNorm):
init.constant_(module.weight, factor * 1.0)
Expand Down