From 12cf40c568e953ea817c071e662b9b42a75df6d4 Mon Sep 17 00:00:00 2001 From: Abhinav Garg Date: Wed, 25 Feb 2026 12:04:52 -0800 Subject: [PATCH 1/3] Update README to clarify section title from "Overview" to "(Testing) Overview" --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index bc6efef9..a8eb5804 100644 --- a/README.md +++ b/README.md @@ -15,7 +15,7 @@ -## Overview +## (Testing) Overview NeMo DFM (Diffusion Foundation Models) is a library under [NeMo Framework](https://github.com/NVIDIA-NeMo), focusing on diffusion models for **Video**, **Image**, and **Text** generation. It unifies cutting-edge diffusion-based architectures and training techniques, prioritizing efficiency and performance from research prototyping to production deployment. From f43b1421ee8941317bf79da425eddb05024d3ab3 Mon Sep 17 00:00:00 2001 From: Abhinav Garg Date: Wed, 25 Feb 2026 12:10:45 -0800 Subject: [PATCH 2/3] Update Wan3DRopeEmbeddings docstring to clarify purpose for testing. --- dfm/src/megatron/model/wan/rope_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dfm/src/megatron/model/wan/rope_utils.py b/dfm/src/megatron/model/wan/rope_utils.py index e3449275..4e187fb5 100644 --- a/dfm/src/megatron/model/wan/rope_utils.py +++ b/dfm/src/megatron/model/wan/rope_utils.py @@ -18,7 +18,7 @@ class Wan3DRopeEmbeddings(torch.nn.Module): """ - Wan 3D RoPE embeddings implementation. + Wan 3D RoPE embeddings implementation. Change for Testing. Implements Wan's 3D RoPE embeddings for Mcore Attention based on Wan's implementation at https://github.com/Wan-Video/Wan2.1/blob/main/wan/modules/model.py. """ From 013cecae5792a0ed0270676ca47865bc8175e8ff Mon Sep 17 00:00:00 2001 From: Abhinav Garg Date: Wed, 25 Feb 2026 12:11:51 -0800 Subject: [PATCH 3/3] Update docstring in Wan3DRopeEmbeddings to maintain consistency in formatting. --- dfm/src/megatron/model/wan/rope_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dfm/src/megatron/model/wan/rope_utils.py b/dfm/src/megatron/model/wan/rope_utils.py index 4e187fb5..1fe4ae80 100644 --- a/dfm/src/megatron/model/wan/rope_utils.py +++ b/dfm/src/megatron/model/wan/rope_utils.py @@ -18,7 +18,7 @@ class Wan3DRopeEmbeddings(torch.nn.Module): """ - Wan 3D RoPE embeddings implementation. Change for Testing. + Wan 3D RoPE embeddings implementation. Change for Testing. Implements Wan's 3D RoPE embeddings for Mcore Attention based on Wan's implementation at https://github.com/Wan-Video/Wan2.1/blob/main/wan/modules/model.py. """