NVIDIA
where is nemo/collections/diffusion/train.py?
i do not find this file, torchrun --nproc-per-node 8 nemo/collections/diffusion/train.py --yes --factory pretrain_xl
This is an AI-generated response. Please verify.
The file "nemo/collections/diffusion/train.py" mentioned in the command appears to be missing from the repository. While the documentation in the diffusion module's readme.rst references this file, it doesn't actually exist in the codebase.
The repository does contain diffusion model training capabilities through recipes like flux_12b.py and flux_535m.py in the nemo/collections/diffusion/recipes directory, but the main train.py script and the "pretrain_xl" factory mentioned in your command are not present.
You might need to check if there's a newer version of the repository that includes this file, or contact the repository maintainers to clarify how to run diffusion model training without this missing file.
I have the same problem. @abhinavg4 Could you help us to figure out it. Great thanks!
it's deprecated :'(
https://github.com/NVIDIA-NeMo/NeMo/blob/main/nemo/collections/diffusion/init.py#L18C1-L22C2
are you going to include diffusion in https://github.com/NVIDIA-NeMo/Megatron-Bridge ?
it actually was displaced here : https://github.com/NVIDIA-NeMo/NeMo/blob/main/scripts/dit/dit_train.py (https://github.com/NVIDIA-NeMo/NeMo/pull/11794/files#diff-e8f95c69b28e59168ac4dc806216eb2d1b37de017cb5c70b74809e3def926c4bL17)