diff --git a/go1/internvl/model/go1/configuration_go1.py b/go1/internvl/model/go1/configuration_go1.py index 9777311..18ab76a 100644 --- a/go1/internvl/model/go1/configuration_go1.py +++ b/go1/internvl/model/go1/configuration_go1.py @@ -97,6 +97,7 @@ def __init__( self.norm = norm self.latent_planning = latent_planning self.action_chunk_size = self.action_config.action_chunk_size + self.initializer_range = 0.02 def to_dict(self): """