Skip to content

eliminate megatron/model/__init__.py to avoid circular import#192

Merged
stas00 merged 1 commit intomainfrom
init-py
Nov 19, 2021
Merged

eliminate megatron/model/__init__.py to avoid circular import#192
stas00 merged 1 commit intomainfrom
init-py

Conversation

@stas00
Copy link
Contributor

@stas00 stas00 commented Nov 19, 2021

replaying #117 as I can't push into that branch to rebase it

We are trying to eliminate this issue:

ImportError: cannot import name 'print_rank_0' from partially initialized module 'megatron' 
(most likely due to a circular import) (.../megatron/__init__.py)

Fixes: #73

@stas00 stas00 changed the title removed regular package for megatron model eliminate megatron/model/__init__.py to avoid circular import Nov 19, 2021
@stas00 stas00 merged commit 4989bcd into main Nov 19, 2021
@stas00 stas00 deleted the init-py branch November 19, 2021 05:29
adammoody pushed a commit to adammoody/Megatron-DeepSpeed that referenced this pull request Aug 16, 2023
* add back local_rank

* fix MoE model init

* fix MoE
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.

eliminate megatron/model/__init__.py

1 participant