Skip to content

Conversation

@nguyen599
Copy link
Contributor

@nguyen599 nguyen599 commented Sep 24, 2025

This PR just fixes tiny error for pr 7559 in the comment reported error here.

[rank1]:   File "/usr/local/lib/python3.11/dist-packages/deepspeed/runtime/engine.py", line 1462, in _configure_optimizer
[rank1]:     self.optimizer = self._configure_zero_optimizer(basic_optimizer)
[rank1]:                      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank1]:   File "/usr/local/lib/python3.11/dist-packages/deepspeed/runtime/engine.py", line 1835, in _configure_zero_optimizer
[rank1]:     from deepspeed.runtime.superoffload.superoffload_stage3 import SuperOffloadOptimizer_Stage3
[rank1]: ModuleNotFoundError: No module named 'deepspeed.runtime.superoffload'

Create __init__.py for superoffload folder to avoid import error when superoffload folder irgnored by pip installation.

Signed-off-by: nguyen599 <[email protected]>
Signed-off-by: nguyen599 <[email protected]>
@xylian86
Copy link
Contributor

@sfc-gh-truwase LGTM

@sfc-gh-truwase sfc-gh-truwase enabled auto-merge (squash) September 24, 2025 16:28
@sfc-gh-truwase sfc-gh-truwase merged commit 16c1bf4 into deepspeedai:master Sep 24, 2025
11 of 12 checks passed
mauryaavinash95 pushed a commit to DataStates/DeepSpeed that referenced this pull request Oct 4, 2025
This PR just fixes tiny error for pr
[7559](deepspeedai#7559) in the
comment reported error
[here](deepspeedai#7559 (comment)).

```
[rank1]:   File "/usr/local/lib/python3.11/dist-packages/deepspeed/runtime/engine.py", line 1462, in _configure_optimizer
[rank1]:     self.optimizer = self._configure_zero_optimizer(basic_optimizer)
[rank1]:                      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
[rank1]:   File "/usr/local/lib/python3.11/dist-packages/deepspeed/runtime/engine.py", line 1835, in _configure_zero_optimizer
[rank1]:     from deepspeed.runtime.superoffload.superoffload_stage3 import SuperOffloadOptimizer_Stage3
[rank1]: ModuleNotFoundError: No module named 'deepspeed.runtime.superoffload'
```

Create `__init__.py` for superoffload folder to avoid import error when
superoffload folder irgnored by pip installation.

---------

Signed-off-by: nguyen599 <[email protected]>
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.

3 participants