Important

You are viewing the NeMo 2.0 documentation. This release introduces significant changes to the API and a new library, NeMo Run. We are currently porting all features from NeMo 1.0 to 2.0. For documentation on previous versions or features not yet available in 2.0, please refer to the NeMo 24.07 documentation.

Diffusion Models

NeMo 2.0 supports training diffusion models, including text-to-image and text-to-video models. NeMo 2.0 uses NeMo-Run to facilitate scaling across multiple GPUs. The following diffusion models are currently supported in NeMo 2.0:

• Flux
• Diffusion Training Framework

Default configurations are provided for each model and are outlined in the model-specific documentation linked above. Every configuration can be modified to train on new datasets or to test new model hyperparameters.