Nemo framework error

[mariamhegazy]

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.

[ethanhe42]

you can try build container with this Dockerfile

FROM nvcr.io/nvidia/nemo:24.12

RUN pip install --no-cache-dir imageio[ffmpeg] pyav iopath better_profanity peft git+https://github.com/NVlabs/Pytorch_Retinaface.git@b843f45

ARG GITLAB_CLONE_ACCESS_TOKEN
RUN rm -rf /opt/megatron-lm
RUN git clone --single-branch --branch cosmos [email protected]:NVIDIA/Megatron-LM.git /opt/megatron-lm && cd /opt/megatron-lm && pip install -e . && rm -rf .git && cd megatron/core/datasets && make
RUN rm -rf /opt/NeMo
RUN git clone --single-branch --branch physicalai [email protected]:NVIDIA/NeMo.git /opt/NeMo && cd /opt/NeMo && git checkout 75aa5fe3e5acc5a885bc3409be553960067fabb9 && rm -rf .git