Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Nemo framework error #77

Open
mariamhegazy opened this issue Jan 21, 2025 · 0 comments
Open

Nemo framework error #77

mariamhegazy opened this issue Jan 21, 2025 · 0 comments
Assignees

Comments

@mariamhegazy
Copy link

mariamhegazy commented Jan 21, 2025

if I want to build my own image instead of using nvcr.io/nvidia/nemo:cosmos.1.0, which branch should I use for installation of the NeMo framework? Following the installation guidelines seems to be missing some modules
ModuleNotFoundError: No module named 'nemo.collections.diffusion'

The main problem is that nvcr.io/nvidia/nemo:cosmos.1.0 only supports amd64 architecture and I want to build it on ARM. Is there a workaround for this ?

Thanks.

@mharrim mharrim assigned mharrim and ethanhe42 and unassigned mharrim Jan 22, 2025
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

No branches or pull requests

3 participants