Skip to content

Fix: add initializer_range default to GO1ModelConfig#156

Open
wadeKeith wants to merge 1 commit intoOpenDriveLab:mainfrom
wadeKeith:fix/initializer-range
Open

Fix: add initializer_range default to GO1ModelConfig#156
wadeKeith wants to merge 1 commit intoOpenDriveLab:mainfrom
wadeKeith:fix/initializer-range

Conversation

@wadeKeith
Copy link
Copy Markdown

Fixes Issue #126

Problem

When loading GO-1 pretrained checkpoint from HuggingFace (agibot-world/GO-1), the model fails during initialization with:

AttributeError: 'GO1ModelConfig' object has no attribute 'initializer_range'

This occurs in modeling_action_expert.py line 513, where _init_weights accesses self.config.initializer_range, but GO1ModelConfig did not have this attribute.

Solution

Add initializer_range = 0.02 as a default attribute in GO1ModelConfig.__init__.

Changes

  • go1/internvl/model/go1/configuration_go1.py: Added self.initializer_range = 0.02

Fixes AttributeError when loading GO-1 pretrained checkpoint from HuggingFace.
The _init_weights method in modeling_action_expert.py accesses self.config.initializer_range,
but GO1ModelConfig did not have this attribute set.

Issue: OpenDriveLab#126
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant