diff --git a/CONTRIBUTING.md b/CONTRIBUTING.md index 3c8e2caf..68ab66d4 100644 --- a/CONTRIBUTING.md +++ b/CONTRIBUTING.md @@ -1,4 +1,4 @@ -# Contributing To NeMo VFM +# Contributing To NeMo DFM ## Signing Your Work diff --git a/README.md b/README.md index 88ade2fb..8213f9f7 100644 --- a/README.md +++ b/README.md @@ -1,14 +1,14 @@ -# NeMo VFM: video foundation model collection +# NeMo DFM: video foundation model collection -NeMo VFM is a state-of-the-art framework for fast, large-scale training and inference of video world models. It unifies the latest diffusion-based and autoregressive techniques, prioritizing efficiency and performance from research prototyping to production deployment. +NeMo DFM is a state-of-the-art framework for fast, large-scale training and inference of video world models. It unifies the latest diffusion-based and autoregressive techniques, prioritizing efficiency and performance from research prototyping to production deployment. ## Projects This collection consists of 4 projects: -1. [Scalable diffusion training framework](nemo_vfm/diffusion/readme.rst) -2. [Accelerated diffusion world models](nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/README.md) -3. [Accelerated autoregressive world models](nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/README.md) -4. [Sparse attention for efficient diffusion inference](nemo_vfm/sparse_attention/README.md) +1. [Scalable diffusion training framework](nemo_dfm/diffusion/readme.rst) +2. [Accelerated diffusion world models](nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/README.md) +3. [Accelerated autoregressive world models](nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/README.md) +4. [Sparse attention for efficient diffusion inference](nemo_dfm/sparse_attention/README.md) ## Citations diff --git a/docs/conf.py b/docs/conf.py index 95e4676e..71c6db07 100644 --- a/docs/conf.py +++ b/docs/conf.py @@ -24,7 +24,7 @@ # -- Project information ----------------------------------------------------- # https://www.sphinx-doc.org/en/master/usage/configuration.html#project-information -project = "NeMo VFM" +project = "NeMo DFM" copyright = "2025, NVIDIA Corporation" author = "NVIDIA Corporation" release = "0.1.0" @@ -68,7 +68,7 @@ sys.path.insert(0, os.path.abspath("..")) autodoc2_packages = [ - "../nemo_vfm", # Path to your package relative to conf.py + "../nemo_dfm", # Path to your package relative to conf.py ] autodoc2_render_plugin = "myst" # Use MyST for rendering docstrings autodoc2_output_dir = "apidocs" # Output directory for autodoc2 (relative to docs/) diff --git a/docs/project.json b/docs/project.json index 90ce11b6..cd3c569b 100644 --- a/docs/project.json +++ b/docs/project.json @@ -1 +1 @@ -{"name": "nemo-vfm", "version": "0.1.0"} +{"name": "nemo-dfm", "version": "0.1.0"} diff --git a/nemo_vfm/__init__.py b/nemo_dfm/__init__.py similarity index 95% rename from nemo_vfm/__init__.py rename to nemo_dfm/__init__.py index 507bd6d0..68d236c5 100644 --- a/nemo_vfm/__init__.py +++ b/nemo_dfm/__init__.py @@ -11,7 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -from nemo_vfm.package_info import ( +from nemo_dfm.package_info import ( __contact_emails__, __contact_names__, __description__, diff --git a/nemo_vfm/diffusion/Dockerfile b/nemo_dfm/diffusion/Dockerfile similarity index 100% rename from nemo_vfm/diffusion/Dockerfile rename to nemo_dfm/diffusion/Dockerfile diff --git a/nemo_vfm/diffusion/__init__.py b/nemo_dfm/diffusion/__init__.py similarity index 100% rename from nemo_vfm/diffusion/__init__.py rename to nemo_dfm/diffusion/__init__.py diff --git a/nemo_vfm/diffusion/assets/mixed_training.png b/nemo_dfm/diffusion/assets/mixed_training.png similarity index 100% rename from nemo_vfm/diffusion/assets/mixed_training.png rename to nemo_dfm/diffusion/assets/mixed_training.png diff --git a/nemo_vfm/diffusion/assets/pipeline_conditioning.png b/nemo_dfm/diffusion/assets/pipeline_conditioning.png similarity index 100% rename from nemo_vfm/diffusion/assets/pipeline_conditioning.png rename to nemo_dfm/diffusion/assets/pipeline_conditioning.png diff --git a/nemo_vfm/diffusion/assets/st_dit_hybrid_parallel.png b/nemo_dfm/diffusion/assets/st_dit_hybrid_parallel.png similarity index 100% rename from nemo_vfm/diffusion/assets/st_dit_hybrid_parallel.png rename to nemo_dfm/diffusion/assets/st_dit_hybrid_parallel.png diff --git a/nemo_vfm/diffusion/data/__init__.py b/nemo_dfm/diffusion/data/__init__.py similarity index 100% rename from nemo_vfm/diffusion/data/__init__.py rename to nemo_dfm/diffusion/data/__init__.py diff --git a/nemo_vfm/diffusion/data/camera.py b/nemo_dfm/diffusion/data/camera.py similarity index 100% rename from nemo_vfm/diffusion/data/camera.py rename to nemo_dfm/diffusion/data/camera.py diff --git a/nemo_vfm/diffusion/data/camera_ctrl_utils.py b/nemo_dfm/diffusion/data/camera_ctrl_utils.py similarity index 100% rename from nemo_vfm/diffusion/data/camera_ctrl_utils.py rename to nemo_dfm/diffusion/data/camera_ctrl_utils.py diff --git a/nemo_vfm/diffusion/data/diffusion_energon_datamodule.py b/nemo_dfm/diffusion/data/diffusion_energon_datamodule.py similarity index 100% rename from nemo_vfm/diffusion/data/diffusion_energon_datamodule.py rename to nemo_dfm/diffusion/data/diffusion_energon_datamodule.py diff --git a/nemo_vfm/diffusion/data/diffusion_fake_datamodule.py b/nemo_dfm/diffusion/data/diffusion_fake_datamodule.py similarity index 100% rename from nemo_vfm/diffusion/data/diffusion_fake_datamodule.py rename to nemo_dfm/diffusion/data/diffusion_fake_datamodule.py diff --git a/nemo_vfm/diffusion/data/diffusion_mock_datamodule.py b/nemo_dfm/diffusion/data/diffusion_mock_datamodule.py similarity index 100% rename from nemo_vfm/diffusion/data/diffusion_mock_datamodule.py rename to nemo_dfm/diffusion/data/diffusion_mock_datamodule.py diff --git a/nemo_vfm/diffusion/data/diffusion_taskencoder.py b/nemo_dfm/diffusion/data/diffusion_taskencoder.py similarity index 100% rename from nemo_vfm/diffusion/data/diffusion_taskencoder.py rename to nemo_dfm/diffusion/data/diffusion_taskencoder.py diff --git a/nemo_vfm/diffusion/data/prepare_energon_dataset.py b/nemo_dfm/diffusion/data/prepare_energon_dataset.py similarity index 100% rename from nemo_vfm/diffusion/data/prepare_energon_dataset.py rename to nemo_dfm/diffusion/data/prepare_energon_dataset.py diff --git a/nemo_vfm/diffusion/data/readme.rst b/nemo_dfm/diffusion/data/readme.rst similarity index 100% rename from nemo_vfm/diffusion/data/readme.rst rename to nemo_dfm/diffusion/data/readme.rst diff --git a/nemo_vfm/diffusion/data/test_datamodule.py b/nemo_dfm/diffusion/data/test_datamodule.py similarity index 100% rename from nemo_vfm/diffusion/data/test_datamodule.py rename to nemo_dfm/diffusion/data/test_datamodule.py diff --git a/nemo_vfm/diffusion/data/utils.py b/nemo_dfm/diffusion/data/utils.py similarity index 100% rename from nemo_vfm/diffusion/data/utils.py rename to nemo_dfm/diffusion/data/utils.py diff --git a/nemo_vfm/diffusion/datamodule.py b/nemo_dfm/diffusion/datamodule.py similarity index 100% rename from nemo_vfm/diffusion/datamodule.py rename to nemo_dfm/diffusion/datamodule.py diff --git a/nemo_vfm/diffusion/encoders/__init__.py b/nemo_dfm/diffusion/encoders/__init__.py similarity index 100% rename from nemo_vfm/diffusion/encoders/__init__.py rename to nemo_dfm/diffusion/encoders/__init__.py diff --git a/nemo_vfm/diffusion/encoders/conditioner.py b/nemo_dfm/diffusion/encoders/conditioner.py similarity index 100% rename from nemo_vfm/diffusion/encoders/conditioner.py rename to nemo_dfm/diffusion/encoders/conditioner.py diff --git a/nemo_vfm/diffusion/models/__init__.py b/nemo_dfm/diffusion/models/__init__.py similarity index 100% rename from nemo_vfm/diffusion/models/__init__.py rename to nemo_dfm/diffusion/models/__init__.py diff --git a/nemo_vfm/diffusion/models/dit/__init__.py b/nemo_dfm/diffusion/models/dit/__init__.py similarity index 100% rename from nemo_vfm/diffusion/models/dit/__init__.py rename to nemo_dfm/diffusion/models/dit/__init__.py diff --git a/nemo_vfm/diffusion/models/dit/action_control/__init__.py b/nemo_dfm/diffusion/models/dit/action_control/__init__.py similarity index 100% rename from nemo_vfm/diffusion/models/dit/action_control/__init__.py rename to nemo_dfm/diffusion/models/dit/action_control/__init__.py diff --git a/nemo_vfm/diffusion/models/dit/action_control/action_control_layers.py b/nemo_dfm/diffusion/models/dit/action_control/action_control_layers.py similarity index 100% rename from nemo_vfm/diffusion/models/dit/action_control/action_control_layers.py rename to nemo_dfm/diffusion/models/dit/action_control/action_control_layers.py diff --git a/nemo_vfm/diffusion/models/dit/cosmos_layer_spec.py b/nemo_dfm/diffusion/models/dit/cosmos_layer_spec.py similarity index 100% rename from nemo_vfm/diffusion/models/dit/cosmos_layer_spec.py rename to nemo_dfm/diffusion/models/dit/cosmos_layer_spec.py diff --git a/nemo_vfm/diffusion/models/dit/dit_attention.py b/nemo_dfm/diffusion/models/dit/dit_attention.py similarity index 100% rename from nemo_vfm/diffusion/models/dit/dit_attention.py rename to nemo_dfm/diffusion/models/dit/dit_attention.py diff --git a/nemo_vfm/diffusion/models/dit/dit_embeddings.py b/nemo_dfm/diffusion/models/dit/dit_embeddings.py similarity index 100% rename from nemo_vfm/diffusion/models/dit/dit_embeddings.py rename to nemo_dfm/diffusion/models/dit/dit_embeddings.py diff --git a/nemo_vfm/diffusion/models/dit/dit_layer_spec.py b/nemo_dfm/diffusion/models/dit/dit_layer_spec.py similarity index 100% rename from nemo_vfm/diffusion/models/dit/dit_layer_spec.py rename to nemo_dfm/diffusion/models/dit/dit_layer_spec.py diff --git a/nemo_vfm/diffusion/models/dit/dit_model.py b/nemo_dfm/diffusion/models/dit/dit_model.py similarity index 100% rename from nemo_vfm/diffusion/models/dit/dit_model.py rename to nemo_dfm/diffusion/models/dit/dit_model.py diff --git a/nemo_vfm/diffusion/models/dit/dit_model_14b.py b/nemo_dfm/diffusion/models/dit/dit_model_14b.py similarity index 100% rename from nemo_vfm/diffusion/models/dit/dit_model_14b.py rename to nemo_dfm/diffusion/models/dit/dit_model_14b.py diff --git a/nemo_vfm/diffusion/models/dit/dit_model_7b.py b/nemo_dfm/diffusion/models/dit/dit_model_7b.py old mode 100755 new mode 100644 similarity index 100% rename from nemo_vfm/diffusion/models/dit/dit_model_7b.py rename to nemo_dfm/diffusion/models/dit/dit_model_7b.py diff --git a/nemo_vfm/diffusion/models/dit_llama/__init__.py b/nemo_dfm/diffusion/models/dit_llama/__init__.py similarity index 100% rename from nemo_vfm/diffusion/models/dit_llama/__init__.py rename to nemo_dfm/diffusion/models/dit_llama/__init__.py diff --git a/nemo_vfm/diffusion/models/dit_llama/dit_llama_layer_spec.py b/nemo_dfm/diffusion/models/dit_llama/dit_llama_layer_spec.py similarity index 100% rename from nemo_vfm/diffusion/models/dit_llama/dit_llama_layer_spec.py rename to nemo_dfm/diffusion/models/dit_llama/dit_llama_layer_spec.py diff --git a/nemo_vfm/diffusion/models/dit_llama/dit_llama_model.py b/nemo_dfm/diffusion/models/dit_llama/dit_llama_model.py similarity index 100% rename from nemo_vfm/diffusion/models/dit_llama/dit_llama_model.py rename to nemo_dfm/diffusion/models/dit_llama/dit_llama_model.py diff --git a/nemo_vfm/diffusion/models/flux/__init__.py b/nemo_dfm/diffusion/models/flux/__init__.py similarity index 100% rename from nemo_vfm/diffusion/models/flux/__init__.py rename to nemo_dfm/diffusion/models/flux/__init__.py diff --git a/nemo_vfm/diffusion/models/flux/layers.py b/nemo_dfm/diffusion/models/flux/layers.py similarity index 100% rename from nemo_vfm/diffusion/models/flux/layers.py rename to nemo_dfm/diffusion/models/flux/layers.py diff --git a/nemo_vfm/diffusion/models/flux/model.py b/nemo_dfm/diffusion/models/flux/model.py similarity index 100% rename from nemo_vfm/diffusion/models/flux/model.py rename to nemo_dfm/diffusion/models/flux/model.py diff --git a/nemo_vfm/diffusion/models/flux/pipeline.py b/nemo_dfm/diffusion/models/flux/pipeline.py similarity index 100% rename from nemo_vfm/diffusion/models/flux/pipeline.py rename to nemo_dfm/diffusion/models/flux/pipeline.py diff --git a/nemo_vfm/diffusion/models/flux_controlnet/__init__.py b/nemo_dfm/diffusion/models/flux_controlnet/__init__.py similarity index 100% rename from nemo_vfm/diffusion/models/flux_controlnet/__init__.py rename to nemo_dfm/diffusion/models/flux_controlnet/__init__.py diff --git a/nemo_vfm/diffusion/models/flux_controlnet/layers.py b/nemo_dfm/diffusion/models/flux_controlnet/layers.py similarity index 100% rename from nemo_vfm/diffusion/models/flux_controlnet/layers.py rename to nemo_dfm/diffusion/models/flux_controlnet/layers.py diff --git a/nemo_vfm/diffusion/models/flux_controlnet/model.py b/nemo_dfm/diffusion/models/flux_controlnet/model.py similarity index 100% rename from nemo_vfm/diffusion/models/flux_controlnet/model.py rename to nemo_dfm/diffusion/models/flux_controlnet/model.py diff --git a/nemo_vfm/diffusion/models/model.py b/nemo_dfm/diffusion/models/model.py old mode 100755 new mode 100644 similarity index 100% rename from nemo_vfm/diffusion/models/model.py rename to nemo_dfm/diffusion/models/model.py diff --git a/nemo_vfm/diffusion/models/moviegen/moviegen_model.py b/nemo_dfm/diffusion/models/moviegen/moviegen_model.py similarity index 100% rename from nemo_vfm/diffusion/models/moviegen/moviegen_model.py rename to nemo_dfm/diffusion/models/moviegen/moviegen_model.py diff --git a/nemo_vfm/diffusion/readme.rst b/nemo_dfm/diffusion/readme.rst similarity index 100% rename from nemo_vfm/diffusion/readme.rst rename to nemo_dfm/diffusion/readme.rst diff --git a/nemo_vfm/diffusion/recipes/__init__.py b/nemo_dfm/diffusion/recipes/__init__.py similarity index 100% rename from nemo_vfm/diffusion/recipes/__init__.py rename to nemo_dfm/diffusion/recipes/__init__.py diff --git a/nemo_vfm/diffusion/recipes/flux_12b.py b/nemo_dfm/diffusion/recipes/flux_12b.py similarity index 100% rename from nemo_vfm/diffusion/recipes/flux_12b.py rename to nemo_dfm/diffusion/recipes/flux_12b.py diff --git a/nemo_vfm/diffusion/recipes/flux_535m.py b/nemo_dfm/diffusion/recipes/flux_535m.py similarity index 100% rename from nemo_vfm/diffusion/recipes/flux_535m.py rename to nemo_dfm/diffusion/recipes/flux_535m.py diff --git a/nemo_vfm/diffusion/sampler/__init__.py b/nemo_dfm/diffusion/sampler/__init__.py similarity index 100% rename from nemo_vfm/diffusion/sampler/__init__.py rename to nemo_dfm/diffusion/sampler/__init__.py diff --git a/nemo_vfm/diffusion/sampler/batch_ops.py b/nemo_dfm/diffusion/sampler/batch_ops.py similarity index 100% rename from nemo_vfm/diffusion/sampler/batch_ops.py rename to nemo_dfm/diffusion/sampler/batch_ops.py diff --git a/nemo_vfm/diffusion/sampler/conditioner.py b/nemo_dfm/diffusion/sampler/conditioner.py old mode 100755 new mode 100644 similarity index 100% rename from nemo_vfm/diffusion/sampler/conditioner.py rename to nemo_dfm/diffusion/sampler/conditioner.py diff --git a/nemo_vfm/diffusion/sampler/conditioner_configs.py b/nemo_dfm/diffusion/sampler/conditioner_configs.py old mode 100755 new mode 100644 similarity index 100% rename from nemo_vfm/diffusion/sampler/conditioner_configs.py rename to nemo_dfm/diffusion/sampler/conditioner_configs.py diff --git a/nemo_vfm/diffusion/sampler/context_parallel.py b/nemo_dfm/diffusion/sampler/context_parallel.py similarity index 100% rename from nemo_vfm/diffusion/sampler/context_parallel.py rename to nemo_dfm/diffusion/sampler/context_parallel.py diff --git a/nemo_vfm/diffusion/sampler/cosmos/__init__.py b/nemo_dfm/diffusion/sampler/cosmos/__init__.py similarity index 100% rename from nemo_vfm/diffusion/sampler/cosmos/__init__.py rename to nemo_dfm/diffusion/sampler/cosmos/__init__.py diff --git a/nemo_vfm/diffusion/sampler/cosmos/cosmos_control_diffusion_pipeline.py b/nemo_dfm/diffusion/sampler/cosmos/cosmos_control_diffusion_pipeline.py similarity index 100% rename from nemo_vfm/diffusion/sampler/cosmos/cosmos_control_diffusion_pipeline.py rename to nemo_dfm/diffusion/sampler/cosmos/cosmos_control_diffusion_pipeline.py diff --git a/nemo_vfm/diffusion/sampler/cosmos/cosmos_diffusion_pipeline.py b/nemo_dfm/diffusion/sampler/cosmos/cosmos_diffusion_pipeline.py similarity index 100% rename from nemo_vfm/diffusion/sampler/cosmos/cosmos_diffusion_pipeline.py rename to nemo_dfm/diffusion/sampler/cosmos/cosmos_diffusion_pipeline.py diff --git a/nemo_vfm/diffusion/sampler/cosmos/cosmos_extended_diffusion_pipeline.py b/nemo_dfm/diffusion/sampler/cosmos/cosmos_extended_diffusion_pipeline.py old mode 100755 new mode 100644 similarity index 100% rename from nemo_vfm/diffusion/sampler/cosmos/cosmos_extended_diffusion_pipeline.py rename to nemo_dfm/diffusion/sampler/cosmos/cosmos_extended_diffusion_pipeline.py diff --git a/nemo_vfm/diffusion/sampler/edm/__init__.py b/nemo_dfm/diffusion/sampler/edm/__init__.py similarity index 100% rename from nemo_vfm/diffusion/sampler/edm/__init__.py rename to nemo_dfm/diffusion/sampler/edm/__init__.py diff --git a/nemo_vfm/diffusion/sampler/edm/edm.py b/nemo_dfm/diffusion/sampler/edm/edm.py similarity index 100% rename from nemo_vfm/diffusion/sampler/edm/edm.py rename to nemo_dfm/diffusion/sampler/edm/edm.py diff --git a/nemo_vfm/diffusion/sampler/edm/edm_pipeline.py b/nemo_dfm/diffusion/sampler/edm/edm_pipeline.py similarity index 100% rename from nemo_vfm/diffusion/sampler/edm/edm_pipeline.py rename to nemo_dfm/diffusion/sampler/edm/edm_pipeline.py diff --git a/nemo_vfm/diffusion/sampler/flow_matching/__init__.py b/nemo_dfm/diffusion/sampler/flow_matching/__init__.py similarity index 100% rename from nemo_vfm/diffusion/sampler/flow_matching/__init__.py rename to nemo_dfm/diffusion/sampler/flow_matching/__init__.py diff --git a/nemo_vfm/diffusion/sampler/flow_matching/flow_match_euler_discrete.py b/nemo_dfm/diffusion/sampler/flow_matching/flow_match_euler_discrete.py similarity index 100% rename from nemo_vfm/diffusion/sampler/flow_matching/flow_match_euler_discrete.py rename to nemo_dfm/diffusion/sampler/flow_matching/flow_match_euler_discrete.py diff --git a/nemo_vfm/diffusion/sampler/res/__init__.py b/nemo_dfm/diffusion/sampler/res/__init__.py similarity index 100% rename from nemo_vfm/diffusion/sampler/res/__init__.py rename to nemo_dfm/diffusion/sampler/res/__init__.py diff --git a/nemo_vfm/diffusion/sampler/res/multi_step.py b/nemo_dfm/diffusion/sampler/res/multi_step.py similarity index 100% rename from nemo_vfm/diffusion/sampler/res/multi_step.py rename to nemo_dfm/diffusion/sampler/res/multi_step.py diff --git a/nemo_vfm/diffusion/sampler/res/res_sampler.py b/nemo_dfm/diffusion/sampler/res/res_sampler.py similarity index 100% rename from nemo_vfm/diffusion/sampler/res/res_sampler.py rename to nemo_dfm/diffusion/sampler/res/res_sampler.py diff --git a/nemo_vfm/diffusion/sampler/res/runge_kutta.py b/nemo_dfm/diffusion/sampler/res/runge_kutta.py similarity index 100% rename from nemo_vfm/diffusion/sampler/res/runge_kutta.py rename to nemo_dfm/diffusion/sampler/res/runge_kutta.py diff --git a/nemo_vfm/diffusion/train.py b/nemo_dfm/diffusion/train.py similarity index 100% rename from nemo_vfm/diffusion/train.py rename to nemo_dfm/diffusion/train.py diff --git a/nemo_vfm/diffusion/utils/__init__.py b/nemo_dfm/diffusion/utils/__init__.py similarity index 100% rename from nemo_vfm/diffusion/utils/__init__.py rename to nemo_dfm/diffusion/utils/__init__.py diff --git a/nemo_vfm/diffusion/utils/flux_ckpt_converter.py b/nemo_dfm/diffusion/utils/flux_ckpt_converter.py similarity index 100% rename from nemo_vfm/diffusion/utils/flux_ckpt_converter.py rename to nemo_dfm/diffusion/utils/flux_ckpt_converter.py diff --git a/nemo_vfm/diffusion/utils/flux_pipeline_utils.py b/nemo_dfm/diffusion/utils/flux_pipeline_utils.py similarity index 100% rename from nemo_vfm/diffusion/utils/flux_pipeline_utils.py rename to nemo_dfm/diffusion/utils/flux_pipeline_utils.py diff --git a/nemo_vfm/diffusion/utils/mcore_parallel_utils.py b/nemo_dfm/diffusion/utils/mcore_parallel_utils.py similarity index 100% rename from nemo_vfm/diffusion/utils/mcore_parallel_utils.py rename to nemo_dfm/diffusion/utils/mcore_parallel_utils.py diff --git a/nemo_vfm/diffusion/vae/__init__.py b/nemo_dfm/diffusion/vae/__init__.py similarity index 100% rename from nemo_vfm/diffusion/vae/__init__.py rename to nemo_dfm/diffusion/vae/__init__.py diff --git a/nemo_vfm/diffusion/vae/autoencoder.py b/nemo_dfm/diffusion/vae/autoencoder.py similarity index 100% rename from nemo_vfm/diffusion/vae/autoencoder.py rename to nemo_dfm/diffusion/vae/autoencoder.py diff --git a/nemo_vfm/diffusion/vae/autovae.py b/nemo_dfm/diffusion/vae/autovae.py similarity index 100% rename from nemo_vfm/diffusion/vae/autovae.py rename to nemo_dfm/diffusion/vae/autovae.py diff --git a/nemo_vfm/diffusion/vae/blocks.py b/nemo_dfm/diffusion/vae/blocks.py similarity index 100% rename from nemo_vfm/diffusion/vae/blocks.py rename to nemo_dfm/diffusion/vae/blocks.py diff --git a/nemo_vfm/diffusion/vae/contperceptual_loss.py b/nemo_dfm/diffusion/vae/contperceptual_loss.py similarity index 100% rename from nemo_vfm/diffusion/vae/contperceptual_loss.py rename to nemo_dfm/diffusion/vae/contperceptual_loss.py diff --git a/nemo_vfm/diffusion/vae/diffusers_vae.py b/nemo_dfm/diffusion/vae/diffusers_vae.py similarity index 100% rename from nemo_vfm/diffusion/vae/diffusers_vae.py rename to nemo_dfm/diffusion/vae/diffusers_vae.py diff --git a/nemo_vfm/diffusion/vae/pretrained_vae.py b/nemo_dfm/diffusion/vae/pretrained_vae.py similarity index 100% rename from nemo_vfm/diffusion/vae/pretrained_vae.py rename to nemo_dfm/diffusion/vae/pretrained_vae.py diff --git a/nemo_vfm/diffusion/vae/readme.rst b/nemo_dfm/diffusion/vae/readme.rst similarity index 100% rename from nemo_vfm/diffusion/vae/readme.rst rename to nemo_dfm/diffusion/vae/readme.rst diff --git a/nemo_vfm/diffusion/vae/test_autovae.py b/nemo_dfm/diffusion/vae/test_autovae.py similarity index 100% rename from nemo_vfm/diffusion/vae/test_autovae.py rename to nemo_dfm/diffusion/vae/test_autovae.py diff --git a/nemo_vfm/diffusion/vae/train_vae.py b/nemo_dfm/diffusion/vae/train_vae.py similarity index 100% rename from nemo_vfm/diffusion/vae/train_vae.py rename to nemo_dfm/diffusion/vae/train_vae.py diff --git a/nemo_vfm/diffusion/vae/train_vae.sh b/nemo_dfm/diffusion/vae/train_vae.sh similarity index 100% rename from nemo_vfm/diffusion/vae/train_vae.sh rename to nemo_dfm/diffusion/vae/train_vae.sh diff --git a/nemo_vfm/diffusion/vae/vae16x/config.json b/nemo_dfm/diffusion/vae/vae16x/config.json similarity index 100% rename from nemo_vfm/diffusion/vae/vae16x/config.json rename to nemo_dfm/diffusion/vae/vae16x/config.json diff --git a/nemo_vfm/diffusion/vae/validate_vae.py b/nemo_dfm/diffusion/vae/validate_vae.py similarity index 100% rename from nemo_vfm/diffusion/vae/validate_vae.py rename to nemo_dfm/diffusion/vae/validate_vae.py diff --git a/nemo_vfm/diffusion/vae/video_vae.py b/nemo_dfm/diffusion/vae/video_vae.py similarity index 100% rename from nemo_vfm/diffusion/vae/video_vae.py rename to nemo_dfm/diffusion/vae/video_vae.py diff --git a/nemo_vfm/package_info.py b/nemo_dfm/package_info.py similarity index 81% rename from nemo_vfm/package_info.py rename to nemo_dfm/package_info.py index bdfa8c57..f7c33b74 100644 --- a/nemo_vfm/package_info.py +++ b/nemo_dfm/package_info.py @@ -24,12 +24,12 @@ __shortversion__ = ".".join(map(str, VERSION[:3])) __version__ = ".".join(map(str, VERSION[:3])) + "".join(VERSION[3:]) -__package_name__ = "nemo_vfm" +__package_name__ = "nemo_dfm" __contact_names__ = "NVIDIA" __contact_emails__ = "nemo-toolkit@nvidia.com" -__homepage__ = "https://github.com/NVIDIA-NeMo/NeMo-VFM" -__repository_url__ = "https://github.com/NVIDIA-NeMo/NeMo-VFM" -__download_url__ = "https://github.com/NVIDIA-NeMo/NeMo-VFM/releases" -__description__ = "NeMo VFM" +__homepage__ = "https://github.com/NVIDIA-NeMo/DFM" +__repository_url__ = "https://github.com/NVIDIA-NeMo/DFM" +__download_url__ = "https://github.com/NVIDIA-NeMo/DFM/releases" +__description__ = "NeMo DFM" __license__ = "Apache2" __keywords__ = "deep learning, machine learning, gpu, NLP, pytorch, torch" diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/POST_TRAINING.md b/nemo_dfm/physicalai/Cosmos/cosmos1/models/POST_TRAINING.md similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/POST_TRAINING.md rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/POST_TRAINING.md diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/README.md b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/README.md similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/README.md rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/README.md diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/base/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/base/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/2285.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/2285.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/2285.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/2285.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/312.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/312.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/312.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/312.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/3266.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/3266.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/3266.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/3266.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/43.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/43.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/43.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/43.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/570.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/570.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/570.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/test/570.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/13109.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/13109.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/13109.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/13109.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/1432.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/1432.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/1432.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/1432.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/19139.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/19139.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/19139.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/19139.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/2277.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/2277.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/2277.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/2277.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/23977.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/23977.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/23977.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/23977.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/346.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/346.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/346.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/346.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/4682.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/4682.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/4682.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/4682.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/5873.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/5873.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/5873.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/5873.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/83.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/83.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/83.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/83.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/9496.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/9496.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/9496.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/train/9496.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/1527.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/1527.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/1527.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/1527.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/199.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/199.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/199.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/199.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/1996.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/1996.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/1996.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/1996.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/2390.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/2390.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/2390.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/2390.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/27.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/27.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/27.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/annotation/val/27.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/2285/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/2285/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/2285/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/2285/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/2285/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/2285/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/2285/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/2285/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/312/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/312/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/312/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/312/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/312/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/312/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/312/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/312/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/3266/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/3266/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/3266/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/3266/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/3266/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/3266/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/3266/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/3266/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/43/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/43/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/43/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/43/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/43/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/43/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/43/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/43/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/570/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/570/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/570/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/570/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/570/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/570/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/570/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/test/570/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/13109/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/13109/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/13109/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/13109/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/13109/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/13109/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/13109/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/13109/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/1432/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/1432/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/1432/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/1432/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/1432/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/1432/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/1432/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/1432/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/19139/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/19139/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/19139/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/19139/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/19139/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/19139/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/19139/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/19139/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/2277/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/2277/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/2277/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/2277/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/2277/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/2277/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/2277/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/2277/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/23977/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/23977/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/23977/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/23977/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/23977/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/23977/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/23977/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/23977/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/346/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/346/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/346/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/346/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/346/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/346/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/346/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/346/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/4682/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/4682/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/4682/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/4682/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/4682/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/4682/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/4682/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/4682/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/5873/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/5873/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/5873/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/5873/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/5873/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/5873/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/5873/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/5873/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/83/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/83/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/83/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/83/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/83/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/83/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/83/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/83/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/9496/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/9496/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/9496/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/9496/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/9496/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/9496/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/9496/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/train/9496/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/1527/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/1527/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/1527/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/1527/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/1527/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/1527/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/1527/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/1527/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/199/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/199/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/199/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/199/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/199/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/199/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/199/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/199/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/1996/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/1996/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/1996/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/1996/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/1996/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/1996/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/1996/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/1996/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/2390/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/2390/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/2390/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/2390/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/2390/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/2390/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/2390/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/2390/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/27/rgb.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/27/rgb.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/27/rgb.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/27/rgb.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/27/state.npy b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/27/state.npy similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/27/state.npy rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/bridge/videos/val/27/state.npy diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/nemo/finetuned_result.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/nemo/finetuned_result.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/nemo/finetuned_result.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/nemo/finetuned_result.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/0.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/0.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/0.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/0.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/1.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/1.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/1.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/1.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/2.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/2.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/2.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/2.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/3.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/3.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/3.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/3.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/4.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/4.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/4.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/4.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/5.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/5.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/5.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/5.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/6.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/6.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/6.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/6.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/7.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/7.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/7.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/7.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/8.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/8.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/8.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/8.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/9.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/9.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/9.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/9.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/base.jsonl b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/base.jsonl similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/base.jsonl rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/base.jsonl diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/video2world.jsonl b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/video2world.jsonl similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/video2world.jsonl rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/batch_inputs/video2world.jsonl diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/input.jpg b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/input.jpg similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/input.jpg rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/input.jpg diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/input.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/input.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/input.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/input.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/output_from_image_input_12b.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/output_from_image_input_12b.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/output_from_image_input_12b.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/output_from_image_input_12b.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/output_from_image_input_13b.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/output_from_image_input_13b.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/output_from_image_input_13b.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/output_from_image_input_13b.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/output_from_video_input_12b.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/output_from_video_input_12b.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/output_from_video_input_12b.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/output_from_video_input_12b.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/output_from_video_input_13b.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/output_from_video_input_13b.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/output_from_video_input_13b.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/assets/v1p0/output_from_video_input_13b.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/inference/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/base/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/inference/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/base/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/base/model.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/base/model.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/base/model.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/base/model.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/base/model_config.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/base/model_config.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/base/model_config.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/base/model_config.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/base/tokenizer.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/base/tokenizer.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/base/tokenizer.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/base/tokenizer.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/inference/inference_config.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/inference/inference_config.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/inference/inference_config.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/configs/inference/inference_config.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/config/base/conditioner.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/config/base/conditioner.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/config/base/conditioner.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/config/base/conditioner.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/config/config_latent_diffusion_decoder.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/config/config_latent_diffusion_decoder.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/config/config_latent_diffusion_decoder.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/config/config_latent_diffusion_decoder.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/config/inference/cosmos_diffusiondecoder_7b.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/config/inference/cosmos_diffusiondecoder_7b.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/config/inference/cosmos_diffusiondecoder_7b.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/config/inference/cosmos_diffusiondecoder_7b.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/config/registry.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/config/registry.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/config/registry.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/config/registry.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/inference.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/inference.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/inference.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/inference.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/model.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/model.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/model.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/model.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/network.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/network.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/network.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/network.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/utils.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/utils.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/utils.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/diffusion_decoder/utils.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/inference/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/inference/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/inference/base.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/inference/base.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/inference/base.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/inference/base.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/inference/video2world.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/inference/video2world.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/inference/video2world.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/inference/video2world.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/inference/world_generation_pipeline.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/inference/world_generation_pipeline.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/inference/world_generation_pipeline.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/inference/world_generation_pipeline.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/model.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/model.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/model.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/model.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/attention.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/attention.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/attention.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/attention.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/embedding.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/embedding.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/embedding.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/embedding.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/mlp.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/mlp.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/mlp.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/mlp.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/mm_projector.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/mm_projector.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/mm_projector.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/mm_projector.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/normalization.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/normalization.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/normalization.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/modules/normalization.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/cosmos.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/cosmos.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/cosmos.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/cosmos.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/cosmos_action_control.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/cosmos_action_control.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/cosmos_action_control.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/cosmos_action_control.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/cosmos_video2world.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/cosmos_video2world.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/cosmos_video2world.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/cosmos_video2world.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/download_autoregressive_nemo.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/download_autoregressive_nemo.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/download_autoregressive_nemo.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/download_autoregressive_nemo.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/README.md b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/README.md similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/README.md rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/README.md diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/action_control_infer.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/action_control_infer.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/action_control_infer.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/action_control_infer.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/general.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/general.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/general.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/general.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/inference_controller.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/inference_controller.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/inference_controller.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/inference_controller.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/video2world.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/video2world.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/video2world.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/inference/video2world.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/README.md b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/README.md similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/README.md rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/README.md diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/README.md b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/README.md similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/README.md rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/README.md diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/action_control_ar_dataset.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/action_control_ar_dataset.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/action_control_ar_dataset.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/action_control_ar_dataset.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/action_control_dataset.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/action_control_dataset.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/action_control_dataset.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/action_control_dataset.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/action_control_finetuning.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/action_control_finetuning.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/action_control_finetuning.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/action_control_finetuning.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/prepare_dataset.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/prepare_dataset.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/prepare_dataset.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/action_control/prepare_dataset.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/general.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/general.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/general.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/general.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/prepare_dataset.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/prepare_dataset.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/prepare_dataset.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/prepare_dataset.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/tokenizer/README.md b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/tokenizer/README.md similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/tokenizer/README.md rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/tokenizer/README.md diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/tokenizer/token_bench.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/tokenizer/token_bench.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/tokenizer/token_bench.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/tokenizer/token_bench.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/video2world_dataset.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/video2world_dataset.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/video2world_dataset.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/video2world_dataset.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/video2world_finetuning.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/video2world_finetuning.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/video2world_finetuning.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/video2world_finetuning.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/video2world_prepare_dataset.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/video2world_prepare_dataset.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/video2world_prepare_dataset.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/post_training/video2world_prepare_dataset.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/utils.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/utils.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/utils.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/nemo/utils.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/networks/transformer.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/networks/transformer.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/networks/transformer.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/networks/transformer.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/networks/vit.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/networks/vit.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/networks/vit.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/networks/vit.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/discrete_video.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/discrete_video.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/discrete_video.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/discrete_video.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/image_text_tokenizer.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/image_text_tokenizer.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/image_text_tokenizer.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/image_text_tokenizer.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/modules.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/modules.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/modules.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/modules.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/networks.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/networks.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/networks.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/networks.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/patching.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/patching.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/patching.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/patching.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/quantizers.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/quantizers.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/quantizers.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/quantizers.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/text_tokenizer.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/text_tokenizer.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/text_tokenizer.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/text_tokenizer.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/tokenizer.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/tokenizer.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/tokenizer.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/tokenizer.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/utils.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/utils.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/utils.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/tokenizer/utils.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/aegis/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/aegis/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/checkpoint.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/checkpoint.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/checkpoint.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/checkpoint.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/inference.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/inference.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/inference.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/inference.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/misc.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/misc.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/misc.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/misc.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/sampling.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/sampling.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/sampling.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/autoregressive/utils/sampling.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/common/base_world_generation_pipeline.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/common/base_world_generation_pipeline.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/common/base_world_generation_pipeline.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/common/base_world_generation_pipeline.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/common/t5_text_encoder.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/common/t5_text_encoder.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/common/t5_text_encoder.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/common/t5_text_encoder.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/README.md b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/README.md similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/README.md rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/README.md diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/nemo/text2world_example_after_finetune.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/nemo/text2world_example_after_finetune.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/nemo/text2world_example_after_finetune.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/nemo/text2world_example_after_finetune.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/nemo_diffusion_example_data/output_Digit_Lift_movie.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/nemo_diffusion_example_data/output_Digit_Lift_movie.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/nemo_diffusion_example_data/output_Digit_Lift_movie.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/nemo_diffusion_example_data/output_Digit_Lift_movie.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/nemo_diffusion_example_data/output_digit_lift.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/nemo_diffusion_example_data/output_digit_lift.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/nemo_diffusion_example_data/output_digit_lift.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/nemo_diffusion_example_data/output_digit_lift.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/batch_inputs/text2world.jsonl b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/batch_inputs/text2world.jsonl similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/batch_inputs/text2world.jsonl rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/batch_inputs/text2world.jsonl diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/batch_inputs/video2world_ps.jsonl b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/batch_inputs/video2world_ps.jsonl similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/batch_inputs/video2world_ps.jsonl rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/batch_inputs/video2world_ps.jsonl diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/batch_inputs/video2world_wo_ps.jsonl b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/batch_inputs/video2world_wo_ps.jsonl similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/batch_inputs/video2world_wo_ps.jsonl rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/batch_inputs/video2world_wo_ps.jsonl diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/text2world_example.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/text2world_example.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/text2world_example.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/text2world_example.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/text2world_example2.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/text2world_example2.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/text2world_example2.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/text2world_example2.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/video2world_example.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/video2world_example.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/video2world_example.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/video2world_example.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/video2world_input0.jpg b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/video2world_input0.jpg similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/video2world_input0.jpg rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/video2world_input0.jpg diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/video2world_input1.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/video2world_input1.mp4 old mode 100755 new mode 100644 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/video2world_input1.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/video2world_input1.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/video2world_input2.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/video2world_input2.mp4 old mode 100755 new mode 100644 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/video2world_input2.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/assets/v1p0/video2world_input2.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/conditioner.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/conditioner.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/conditioner.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/conditioner.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/conditioner.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/conditioner.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/conditioner.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/conditioner.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/model.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/model.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/model.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/model.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/net.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/net.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/net.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/net.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/tokenizer.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/tokenizer.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/tokenizer.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/base/tokenizer.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/config.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/config.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/config.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/config.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/augmentors.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/augmentors.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/augmentors.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/augmentors.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/blurs.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/blurs.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/blurs.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/blurs.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/conditioner.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/conditioner.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/conditioner.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/conditioner.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/config.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/config.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/config.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/config.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/model.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/model.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/model.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/model.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/net_ctrl.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/net_ctrl.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/net_ctrl.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/net_ctrl.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/registry.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/registry.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/registry.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/ctrl/registry.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/inference/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/inference/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/inference/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/inference/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/inference/cosmos-1-diffusion-text2world.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/inference/cosmos-1-diffusion-text2world.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/inference/cosmos-1-diffusion-text2world.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/inference/cosmos-1-diffusion-text2world.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/inference/cosmos-1-diffusion-video2world.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/inference/cosmos-1-diffusion-video2world.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/inference/cosmos-1-diffusion-video2world.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/inference/cosmos-1-diffusion-video2world.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/inference/cosmos-1_1-diffusion-multi-camera.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/inference/cosmos-1_1-diffusion-multi-camera.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/inference/cosmos-1_1-diffusion-multi-camera.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/inference/cosmos-1_1-diffusion-multi-camera.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/registry.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/registry.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/config/registry.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/config/registry.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/datasets/augmentors/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/datasets/augmentors/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/datasets/augmentors/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/datasets/augmentors/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/datasets/augmentors/control_input.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/datasets/augmentors/control_input.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/datasets/augmentors/control_input.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/datasets/augmentors/control_input.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/datasets/augmentors/guided_filter.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/datasets/augmentors/guided_filter.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/datasets/augmentors/guided_filter.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/datasets/augmentors/guided_filter.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/datasets/augmentors/human_keypoint_utils.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/datasets/augmentors/human_keypoint_utils.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/datasets/augmentors/human_keypoint_utils.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/datasets/augmentors/human_keypoint_utils.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/functional/batch_ops.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/functional/batch_ops.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/functional/batch_ops.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/functional/batch_ops.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/functional/multi_step.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/functional/multi_step.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/functional/multi_step.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/functional/multi_step.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/functional/runge_kutta.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/functional/runge_kutta.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/functional/runge_kutta.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/functional/runge_kutta.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/modules/denoiser_scaling.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/modules/denoiser_scaling.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/modules/denoiser_scaling.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/modules/denoiser_scaling.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/modules/res_sampler.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/modules/res_sampler.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/modules/res_sampler.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/modules/res_sampler.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/types.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/types.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/types.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/diffusion/types.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/inference_utils.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/inference_utils.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/inference_utils.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/inference_utils.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/text2world.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/text2world.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/text2world.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/text2world.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/text2world_sample_multi_view_driving.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/text2world_sample_multi_view_driving.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/text2world_sample_multi_view_driving.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/text2world_sample_multi_view_driving.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/video2world.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/video2world.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/video2world.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/video2world.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/video2world_sample_multi_view_driving.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/video2world_sample_multi_view_driving.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/video2world_sample_multi_view_driving.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/video2world_sample_multi_view_driving.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/world_generation_pipeline.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/world_generation_pipeline.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/world_generation_pipeline.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/inference/world_generation_pipeline.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/model/model_sample_multiview_driving.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/model/model_sample_multiview_driving.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/model/model_sample_multiview_driving.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/model/model_sample_multiview_driving.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/model/model_t2w.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/model/model_t2w.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/model/model_t2w.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/model/model_t2w.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/model/model_v2w.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/model/model_v2w.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/model/model_v2w.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/model/model_v2w.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/module/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/module/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/module/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/module/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/module/attention.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/module/attention.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/module/attention.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/module/attention.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/module/blocks.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/module/blocks.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/module/blocks.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/module/blocks.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/module/position_embedding.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/module/position_embedding.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/module/position_embedding.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/module/position_embedding.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/module/pretrained_vae.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/module/pretrained_vae.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/module/pretrained_vae.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/module/pretrained_vae.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/module/timm.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/module/timm.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/module/timm.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/module/timm.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/download_diffusion_nemo.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/download_diffusion_nemo.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/download_diffusion_nemo.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/download_diffusion_nemo.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/README.md b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/README.md similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/README.md rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/README.md diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/action_control_infer.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/action_control_infer.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/action_control_infer.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/action_control_infer.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/control2world.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/control2world.py similarity index 99% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/control2world.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/control2world.py index 25970308..949fb112 100644 --- a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/control2world.py +++ b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/control2world.py @@ -66,7 +66,7 @@ export CONDITIONED_IMAGE_OR_VIDEO="cosmos1/models/diffusion/assets/v1p0/video2world_input1.mp4" export PROMPT="The video showcases a vibrant green sports car with a sleek design, parked on a concrete surface. The car is oriented towards the right side of the frame, highlighting its rear and side profiles. The backdrop features a picturesque beach, with sandy dunes in the foreground and a serene sea in the background. The sky is clear, displaying a soft gradient of colors that hints at either sunrise or sunset. The scene is tranquil, with no people or other vehicles in sight, and the car remains stationary, inviting viewers to appreciate its beauty against the stunning natural landscape." export CTRL_MODEL_DIR=/lustre/fsw/portfolios/coreai/users/sgovande/models/mcore_blur2world_7B_converted/ctrl -PYTHONPATH=$(pwd):~/lustre/nemo-vfm torchrun --nproc_per_node=$NUM_DEVICES cosmos1/models/diffusion/nemo/inference/control2world.py --cosmos_assets_dir ./checkpoints --conditioned_image_or_video_path $CONDITIONED_IMAGE_OR_VIDEO --num_devices $NUM_DEVICES --cp_size $NUM_DEVICES --prompt "$PROMPT" --control_weight 1 --ctrl_model_dir $CTRL_MODEL_DIR +PYTHONPATH=$(pwd):~/lustre/nemo-dfm torchrun --nproc_per_node=$NUM_DEVICES cosmos1/models/diffusion/nemo/inference/control2world.py --cosmos_assets_dir ./checkpoints --conditioned_image_or_video_path $CONDITIONED_IMAGE_OR_VIDEO --num_devices $NUM_DEVICES --cp_size $NUM_DEVICES --prompt "$PROMPT" --control_weight 1 --ctrl_model_dir $CTRL_MODEL_DIR """ diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/general.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/general.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/general.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/general.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/inference_utils.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/inference_utils.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/inference_utils.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/inference_utils.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/validate_multicamera.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/validate_multicamera.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/validate_multicamera.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/validate_multicamera.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/video2world.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/video2world.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/video2world.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/inference/video2world.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/README.md b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/README.md old mode 100755 new mode 100644 similarity index 99% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/README.md rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/README.md index 6098a289..1dd29667 --- a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/README.md +++ b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/README.md @@ -334,11 +334,11 @@ torchrun --nproc_per_node 8 cosmos1/models/diffusion/nemo/post_training/video2wo To review the complete set of parameters such as the learning rate, checkpointing strategy, model parameters, required video metadata and Tensor `dtype`, and more, the `cosmos_diffusion_{model_size}_video2world_action_ctrl_finetune` NeMo2 recipe and `DiTConfig` are located in these modules: -- NeMo2 Recipe: `nemo-vfm/nemo/collections/diffusion/train.py::finetune_{7b,14b}_action_control` +- NeMo2 Recipe: `nemo-dfm/nemo/collections/diffusion/train.py::finetune_{7b,14b}_action_control` - Training and checkpoint settings. -- `DiTConfig`: `nemo-vfm/nemo/collections/diffusion/models/model.py::DiT{7B,14B}Video2WorldActionConfig` +- `DiTConfig`: `nemo-dfm/nemo/collections/diffusion/models/model.py::DiT{7B,14B}Video2WorldActionConfig` - Model architecture configuration, including the action control vector dimension, e.g. `config.action_emb_dim=7` (i.e. of shape `(B, 7)`) for the Bridge dataset. -- Data Module: `nemo-vfm/nemo/collections/diffusion/datamodule.py::ActionControlDiffusionDataset` +- Data Module: `nemo-dfm/nemo/collections/diffusion/datamodule.py::ActionControlDiffusionDataset` - Action control dataset and data loading. For more information on NeMo2 recipes and NeMo Lightning settings, you can refer to: https://github.com/NVIDIA/NeMo. diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/action_control/README.md b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/action_control/README.md similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/action_control/README.md rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/action_control/README.md diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/action_control/prepare_dataset.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/action_control/prepare_dataset.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/action_control/prepare_dataset.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/action_control/prepare_dataset.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/camera_control/README.md b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/camera_control/README.md similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/camera_control/README.md rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/camera_control/README.md diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/camera_control/prepare_dataset.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/camera_control/prepare_dataset.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/camera_control/prepare_dataset.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/camera_control/prepare_dataset.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/dataverse/av22_tar_index_full.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/dataverse/av22_tar_index_full.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/dataverse/av22_tar_index_full.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/dataverse/av22_tar_index_full.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/general.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/general.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/general.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/general.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/matplotlib/fontlist-v390.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/matplotlib/fontlist-v390.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/matplotlib/fontlist-v390.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/matplotlib/fontlist-v390.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/mapping.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/mapping.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/mapping.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/mapping.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/part_000000/001365.json b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/part_000000/001365.json similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/part_000000/001365.json rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/part_000000/001365.json diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/part_000000/001365.tar b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/part_000000/001365.tar similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/part_000000/001365.tar rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera/AV-V2.2/alpamayo_caption_t5/qwen_t5_tars/trainv2-2-chunk-00/part_000000/001365.tar diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera_mock.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera_mock.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera_mock.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/multicamera_mock.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/prepare_dataset.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/prepare_dataset.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/prepare_dataset.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/prepare_dataset.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/prepare_instruction_dataset.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/prepare_instruction_dataset.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/prepare_instruction_dataset.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/prepare_instruction_dataset.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/torch/megatron/megatron-gpt-345m_merges b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/torch/megatron/megatron-gpt-345m_merges similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/torch/megatron/megatron-gpt-345m_merges rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/torch/megatron/megatron-gpt-345m_merges diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/torch/megatron/megatron-gpt-345m_vocab b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/torch/megatron/megatron-gpt-345m_vocab similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/torch/megatron/megatron-gpt-345m_vocab rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/torch/megatron/megatron-gpt-345m_vocab diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/video2world.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/video2world.py old mode 100755 new mode 100644 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/video2world.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/nemo/post_training/video2world.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/networks/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/networks/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/networks/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/networks/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/networks/general_dit.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/networks/general_dit.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/networks/general_dit.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/networks/general_dit.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/networks/general_dit_multi_camera.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/networks/general_dit_multi_camera.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/networks/general_dit_multi_camera.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/networks/general_dit_multi_camera.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/networks/general_dit_video_conditioned.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/networks/general_dit_video_conditioned.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/networks/general_dit_video_conditioned.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/networks/general_dit_video_conditioned.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/prompt_upsampler/inference.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/prompt_upsampler/inference.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/prompt_upsampler/inference.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/prompt_upsampler/inference.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/prompt_upsampler/text2world_prompt_upsampler_inference.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/prompt_upsampler/text2world_prompt_upsampler_inference.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/prompt_upsampler/text2world_prompt_upsampler_inference.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/prompt_upsampler/text2world_prompt_upsampler_inference.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/prompt_upsampler/video2world_prompt_upsampler_inference.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/prompt_upsampler/video2world_prompt_upsampler_inference.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/diffusion/prompt_upsampler/video2world_prompt_upsampler_inference.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/diffusion/prompt_upsampler/video2world_prompt_upsampler_inference.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/README.md b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/README.md similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/README.md rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/README.md diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/blocklist/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/blocklist/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/face_blur_filter/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/aegis/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/face_blur_filter/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/aegis/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/aegis/aegis.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/aegis/aegis.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/aegis/aegis.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/aegis/aegis.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/aegis/categories.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/aegis/categories.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/aegis/categories.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/aegis/categories.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/video_content_safety_filter/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/blocklist/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/video_content_safety_filter/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/blocklist/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/blocklist/blocklist.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/blocklist/blocklist.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/blocklist/blocklist.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/blocklist/blocklist.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/blocklist/utils.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/blocklist/utils.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/blocklist/utils.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/blocklist/utils.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/common/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/common/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/common/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/common/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/common/core.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/common/core.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/common/core.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/common/core.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/common/io_utils.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/common/io_utils.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/common/io_utils.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/common/io_utils.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/common/presets.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/common/presets.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/common/presets.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/common/presets.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/face_blur_filter/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/face_blur_filter/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/face_blur_filter/blur_utils.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/face_blur_filter/blur_utils.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/face_blur_filter/blur_utils.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/face_blur_filter/blur_utils.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/face_blur_filter/face_blur_filter.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/face_blur_filter/face_blur_filter.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/face_blur_filter/face_blur_filter.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/face_blur_filter/face_blur_filter.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/face_blur_filter/retinaface_utils.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/face_blur_filter/retinaface_utils.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/face_blur_filter/retinaface_utils.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/face_blur_filter/retinaface_utils.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/video_content_safety_filter/__init__.py similarity index 97% rename from nemo_vfm/physicalai/Cosmos/cosmos1/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/video_content_safety_filter/__init__.py index dac9a4d7..3159bfe6 100644 --- a/nemo_vfm/physicalai/Cosmos/cosmos1/__init__.py +++ b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/video_content_safety_filter/__init__.py @@ -1,14 +1,14 @@ -# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. -# SPDX-License-Identifier: Apache-2.0 -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. +# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-License-Identifier: Apache-2.0 +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/video_content_safety_filter/model.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/video_content_safety_filter/model.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/video_content_safety_filter/model.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/video_content_safety_filter/model.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/video_content_safety_filter/video_content_safety_filter.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/video_content_safety_filter/video_content_safety_filter.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/video_content_safety_filter/video_content_safety_filter.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/video_content_safety_filter/video_content_safety_filter.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/video_content_safety_filter/vision_encoder.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/video_content_safety_filter/vision_encoder.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/guardrail/video_content_safety_filter/vision_encoder.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/guardrail/video_content_safety_filter/vision_encoder.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/README.md b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/README.md similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/README.md rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/README.md diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/assets/Davis-results.jpg b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/assets/Davis-results.jpg similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/assets/Davis-results.jpg rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/assets/Davis-results.jpg diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/assets/Performance.jpg b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/assets/Performance.jpg similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/assets/Performance.jpg rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/assets/Performance.jpg diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/assets/arch_diagram.jpg b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/assets/arch_diagram.jpg similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/assets/arch_diagram.jpg rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/assets/arch_diagram.jpg diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/assets/cosmos-tokenizer.jpg b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/assets/cosmos-tokenizer.jpg similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/assets/cosmos-tokenizer.jpg rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/assets/cosmos-tokenizer.jpg diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/__init__.py similarity index 97% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/__init__.py index dac9a4d7..3159bfe6 100644 --- a/nemo_vfm/physicalai/Cosmos/cosmos1/models/__init__.py +++ b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/__init__.py @@ -1,14 +1,14 @@ -# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. -# SPDX-License-Identifier: Apache-2.0 -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. +# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-License-Identifier: Apache-2.0 +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/image_cli.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/image_cli.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/image_cli.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/image_cli.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/image_lib.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/image_lib.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/image_lib.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/image_lib.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/utils.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/utils.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/utils.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/utils.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/video_cli.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/video_cli.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/video_cli.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/video_cli.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/video_lib.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/video_lib.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/video_lib.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/inference/video_lib.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/distributions.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/distributions.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/distributions.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/distributions.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/layers2d.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/layers2d.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/layers2d.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/layers2d.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/layers2d_test.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/layers2d_test.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/layers2d_test.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/layers2d_test.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/layers3d.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/layers3d.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/layers3d.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/layers3d.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/layers3d_test.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/layers3d_test.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/layers3d_test.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/layers3d_test.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/patching.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/patching.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/patching.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/patching.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/quantizers.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/quantizers.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/quantizers.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/quantizers.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/utils.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/utils.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/utils.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/modules/utils.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/nemo/README.md b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/nemo/README.md similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/nemo/README.md rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/nemo/README.md diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/nemo/assets/loss.png b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/nemo/assets/loss.png similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/nemo/assets/loss.png rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/nemo/assets/loss.png diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/nemo/requirements.txt b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/nemo/requirements.txt similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/nemo/requirements.txt rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/nemo/requirements.txt diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/nemo/train_tokenizer.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/nemo/train_tokenizer.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/nemo/train_tokenizer.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/nemo/train_tokenizer.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/configs.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/configs.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/configs.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/configs.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/continuous_image.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/continuous_image.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/continuous_image.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/continuous_image.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/continuous_video.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/continuous_video.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/continuous_video.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/continuous_video.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/discrete_image.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/discrete_image.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/discrete_image.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/discrete_image.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/discrete_video.py b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/discrete_video.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/discrete_video.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/networks/discrete_video.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/notebook/Image_Tokenization.ipynb b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/notebook/Image_Tokenization.ipynb similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/notebook/Image_Tokenization.ipynb rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/notebook/Image_Tokenization.ipynb diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/test_data/image.png b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/test_data/image.png similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/test_data/image.png rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/test_data/image.png diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/test_data/video.mp4 b/nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/test_data/video.mp4 similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/models/tokenizer/test_data/video.mp4 rename to nemo_dfm/physicalai/Cosmos/cosmos1/models/tokenizer/test_data/video.mp4 diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/utils/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/utils/__init__.py similarity index 97% rename from nemo_vfm/physicalai/Cosmos/cosmos1/utils/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/utils/__init__.py index dac9a4d7..3159bfe6 100644 --- a/nemo_vfm/physicalai/Cosmos/cosmos1/utils/__init__.py +++ b/nemo_dfm/physicalai/Cosmos/cosmos1/utils/__init__.py @@ -1,14 +1,14 @@ -# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. -# SPDX-License-Identifier: Apache-2.0 -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. +# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. +# SPDX-License-Identifier: Apache-2.0 +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/utils/config.py b/nemo_dfm/physicalai/Cosmos/cosmos1/utils/config.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/utils/config.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/utils/config.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/utils/config_helper.py b/nemo_dfm/physicalai/Cosmos/cosmos1/utils/config_helper.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/utils/config_helper.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/utils/config_helper.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/utils/device.py b/nemo_dfm/physicalai/Cosmos/cosmos1/utils/device.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/utils/device.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/utils/device.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/utils/distributed.py b/nemo_dfm/physicalai/Cosmos/cosmos1/utils/distributed.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/utils/distributed.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/utils/distributed.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/utils/io.py b/nemo_dfm/physicalai/Cosmos/cosmos1/utils/io.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/utils/io.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/utils/io.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/utils/lazy_config/__init__.py b/nemo_dfm/physicalai/Cosmos/cosmos1/utils/lazy_config/__init__.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/utils/lazy_config/__init__.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/utils/lazy_config/__init__.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/utils/lazy_config/file_io.py b/nemo_dfm/physicalai/Cosmos/cosmos1/utils/lazy_config/file_io.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/utils/lazy_config/file_io.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/utils/lazy_config/file_io.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/utils/lazy_config/instantiate.py b/nemo_dfm/physicalai/Cosmos/cosmos1/utils/lazy_config/instantiate.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/utils/lazy_config/instantiate.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/utils/lazy_config/instantiate.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/utils/lazy_config/lazy.py b/nemo_dfm/physicalai/Cosmos/cosmos1/utils/lazy_config/lazy.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/utils/lazy_config/lazy.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/utils/lazy_config/lazy.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/utils/lazy_config/omegaconf_patch.py b/nemo_dfm/physicalai/Cosmos/cosmos1/utils/lazy_config/omegaconf_patch.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/utils/lazy_config/omegaconf_patch.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/utils/lazy_config/omegaconf_patch.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/utils/lazy_config/registry.py b/nemo_dfm/physicalai/Cosmos/cosmos1/utils/lazy_config/registry.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/utils/lazy_config/registry.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/utils/lazy_config/registry.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/utils/log.py b/nemo_dfm/physicalai/Cosmos/cosmos1/utils/log.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/utils/log.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/utils/log.py diff --git a/nemo_vfm/physicalai/Cosmos/cosmos1/utils/misc.py b/nemo_dfm/physicalai/Cosmos/cosmos1/utils/misc.py similarity index 100% rename from nemo_vfm/physicalai/Cosmos/cosmos1/utils/misc.py rename to nemo_dfm/physicalai/Cosmos/cosmos1/utils/misc.py diff --git a/nemo_vfm/physicalai/Dockerfile b/nemo_dfm/physicalai/Dockerfile similarity index 100% rename from nemo_vfm/physicalai/Dockerfile rename to nemo_dfm/physicalai/Dockerfile diff --git a/nemo_vfm/physicalai/__init__.py b/nemo_dfm/physicalai/__init__.py similarity index 100% rename from nemo_vfm/physicalai/__init__.py rename to nemo_dfm/physicalai/__init__.py diff --git a/nemo_vfm/physicalai/datasets/__init__.py b/nemo_dfm/physicalai/datasets/__init__.py similarity index 100% rename from nemo_vfm/physicalai/datasets/__init__.py rename to nemo_dfm/physicalai/datasets/__init__.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/.gitignore b/nemo_dfm/physicalai/datasets/dataverse/.gitignore similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/.gitignore rename to nemo_dfm/physicalai/datasets/dataverse/.gitignore diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/__init__.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/__init__.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/__init__.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/__init__.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/datasets/__init__.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/datasets/__init__.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/datasets/__init__.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/datasets/__init__.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/datasets/alpamayo_v1.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/datasets/alpamayo_v1.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/datasets/alpamayo_v1.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/datasets/alpamayo_v1.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/datasets/alpamayo_v2.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/datasets/alpamayo_v2.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/datasets/alpamayo_v2.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/datasets/alpamayo_v2.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/datasets/base.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/datasets/base.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/datasets/base.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/datasets/base.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/datasets/cosmos_av_w_traj.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/datasets/cosmos_av_w_traj.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/datasets/cosmos_av_w_traj.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/datasets/cosmos_av_w_traj.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/__init__.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/__init__.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/__init__.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/__init__.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/camera.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/camera.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/camera.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/camera.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/constants.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/constants.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/constants.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/constants.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/egomotion_decoder.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/egomotion_decoder.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/egomotion_decoder.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/egomotion_decoder.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/generator_config.yaml b/nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/generator_config.yaml similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/generator_config.yaml rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/generator_config.yaml diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/intrinsic_decoder.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/intrinsic_decoder.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/intrinsic_decoder.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/intrinsic_decoder.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/ndas_camera_model.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/ndas_camera_model.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/ndas_camera_model.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/ndas_camera_model.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/rig_decoder.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/rig_decoder.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/rig_decoder.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/rig_decoder.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/rotation.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/rotation.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/rotation.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/rotation.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/transformation.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/transformation.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/transformation.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/transformation.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/vis_trajectory.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/vis_trajectory.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/vis_trajectory.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/alpamayo/vis_trajectory.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/colmap.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/colmap.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/colmap.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/colmap.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/general.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/general.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/general.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/general.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/ndas/av_metadata.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/ndas/av_metadata.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/ndas/av_metadata.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/ndas/av_metadata.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/ndas/camera_model.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/ndas/camera_model.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/ndas/camera_model.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/ndas/camera_model.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/ndas/rig.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/ndas/rig.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/ndas/rig.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/ndas/rig.py diff --git a/nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/version.py b/nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/version.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse/dataverse/utils/version.py rename to nemo_dfm/physicalai/datasets/dataverse/dataverse/utils/version.py diff --git a/nemo_vfm/physicalai/datasets/dataverse_dataset/__init__.py b/nemo_dfm/physicalai/datasets/dataverse_dataset/__init__.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse_dataset/__init__.py rename to nemo_dfm/physicalai/datasets/dataverse_dataset/__init__.py diff --git a/nemo_vfm/physicalai/datasets/dataverse_dataset/driving_dataloader/__init__.py b/nemo_dfm/physicalai/datasets/dataverse_dataset/driving_dataloader/__init__.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse_dataset/driving_dataloader/__init__.py rename to nemo_dfm/physicalai/datasets/dataverse_dataset/driving_dataloader/__init__.py diff --git a/nemo_vfm/physicalai/datasets/dataverse_dataset/driving_dataloader/alpamayo_dataloader.py b/nemo_dfm/physicalai/datasets/dataverse_dataset/driving_dataloader/alpamayo_dataloader.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse_dataset/driving_dataloader/alpamayo_dataloader.py rename to nemo_dfm/physicalai/datasets/dataverse_dataset/driving_dataloader/alpamayo_dataloader.py diff --git a/nemo_vfm/physicalai/datasets/dataverse_dataset/driving_dataloader/config_dataverse.py b/nemo_dfm/physicalai/datasets/dataverse_dataset/driving_dataloader/config_dataverse.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse_dataset/driving_dataloader/config_dataverse.py rename to nemo_dfm/physicalai/datasets/dataverse_dataset/driving_dataloader/config_dataverse.py diff --git a/nemo_vfm/physicalai/datasets/dataverse_dataset/driving_dataloader/dataloader_utils.py b/nemo_dfm/physicalai/datasets/dataverse_dataset/driving_dataloader/dataloader_utils.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse_dataset/driving_dataloader/dataloader_utils.py rename to nemo_dfm/physicalai/datasets/dataverse_dataset/driving_dataloader/dataloader_utils.py diff --git a/nemo_vfm/physicalai/datasets/dataverse_dataset/instantiate_utils.py b/nemo_dfm/physicalai/datasets/dataverse_dataset/instantiate_utils.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse_dataset/instantiate_utils.py rename to nemo_dfm/physicalai/datasets/dataverse_dataset/instantiate_utils.py diff --git a/nemo_vfm/physicalai/datasets/dataverse_dataset/schema.py b/nemo_dfm/physicalai/datasets/dataverse_dataset/schema.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse_dataset/schema.py rename to nemo_dfm/physicalai/datasets/dataverse_dataset/schema.py diff --git a/nemo_vfm/physicalai/datasets/dataverse_dataset/utils.py b/nemo_dfm/physicalai/datasets/dataverse_dataset/utils.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse_dataset/utils.py rename to nemo_dfm/physicalai/datasets/dataverse_dataset/utils.py diff --git a/nemo_vfm/physicalai/datasets/dataverse_dataset/video_size.py b/nemo_dfm/physicalai/datasets/dataverse_dataset/video_size.py similarity index 100% rename from nemo_vfm/physicalai/datasets/dataverse_dataset/video_size.py rename to nemo_dfm/physicalai/datasets/dataverse_dataset/video_size.py diff --git a/nemo_vfm/physicalai/datasets/mock_dataset.py b/nemo_dfm/physicalai/datasets/mock_dataset.py similarity index 100% rename from nemo_vfm/physicalai/datasets/mock_dataset.py rename to nemo_dfm/physicalai/datasets/mock_dataset.py diff --git a/nemo_vfm/physicalai/diffusion/post_training/multicamera/README.md b/nemo_dfm/physicalai/diffusion/post_training/multicamera/README.md similarity index 100% rename from nemo_vfm/physicalai/diffusion/post_training/multicamera/README.md rename to nemo_dfm/physicalai/diffusion/post_training/multicamera/README.md diff --git a/nemo_vfm/physicalai/diffusion/post_training/multicamera/dit_multi_camera.py b/nemo_dfm/physicalai/diffusion/post_training/multicamera/dit_multi_camera.py similarity index 100% rename from nemo_vfm/physicalai/diffusion/post_training/multicamera/dit_multi_camera.py rename to nemo_dfm/physicalai/diffusion/post_training/multicamera/dit_multi_camera.py diff --git a/nemo_vfm/physicalai/diffusion/post_training/multicamera/multicamera.py b/nemo_dfm/physicalai/diffusion/post_training/multicamera/multicamera.py similarity index 100% rename from nemo_vfm/physicalai/diffusion/post_training/multicamera/multicamera.py rename to nemo_dfm/physicalai/diffusion/post_training/multicamera/multicamera.py diff --git a/nemo_vfm/physicalai/diffusion/post_training/multicamera/multicamera_mock.py b/nemo_dfm/physicalai/diffusion/post_training/multicamera/multicamera_mock.py similarity index 100% rename from nemo_vfm/physicalai/diffusion/post_training/multicamera/multicamera_mock.py rename to nemo_dfm/physicalai/diffusion/post_training/multicamera/multicamera_mock.py diff --git a/nemo_vfm/physicalai/diffusion/post_training/multicamera/test_dataloader.py b/nemo_dfm/physicalai/diffusion/post_training/multicamera/test_dataloader.py similarity index 100% rename from nemo_vfm/physicalai/diffusion/post_training/multicamera/test_dataloader.py rename to nemo_dfm/physicalai/diffusion/post_training/multicamera/test_dataloader.py diff --git a/nemo_vfm/physicalai/diffusion/post_training/multicamera/train.sh b/nemo_dfm/physicalai/diffusion/post_training/multicamera/train.sh similarity index 100% rename from nemo_vfm/physicalai/diffusion/post_training/multicamera/train.sh rename to nemo_dfm/physicalai/diffusion/post_training/multicamera/train.sh diff --git a/nemo_vfm/physicalai/diffusion/post_training/multicamera/validate_multicamera.py b/nemo_dfm/physicalai/diffusion/post_training/multicamera/validate_multicamera.py similarity index 100% rename from nemo_vfm/physicalai/diffusion/post_training/multicamera/validate_multicamera.py rename to nemo_dfm/physicalai/diffusion/post_training/multicamera/validate_multicamera.py diff --git a/nemo_vfm/physicalai/tokenizer/README.md b/nemo_dfm/physicalai/tokenizer/README.md similarity index 100% rename from nemo_vfm/physicalai/tokenizer/README.md rename to nemo_dfm/physicalai/tokenizer/README.md diff --git a/nemo_vfm/physicalai/tokenizer/assets/loss.png b/nemo_dfm/physicalai/tokenizer/assets/loss.png similarity index 100% rename from nemo_vfm/physicalai/tokenizer/assets/loss.png rename to nemo_dfm/physicalai/tokenizer/assets/loss.png diff --git a/nemo_vfm/physicalai/tokenizer/augmentors.py b/nemo_dfm/physicalai/tokenizer/augmentors.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/augmentors.py rename to nemo_dfm/physicalai/tokenizer/augmentors.py diff --git a/nemo_vfm/physicalai/tokenizer/configs.py b/nemo_dfm/physicalai/tokenizer/configs.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/configs.py rename to nemo_dfm/physicalai/tokenizer/configs.py diff --git a/nemo_vfm/physicalai/tokenizer/data/__init__.py b/nemo_dfm/physicalai/tokenizer/data/__init__.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/data/__init__.py rename to nemo_dfm/physicalai/tokenizer/data/__init__.py diff --git a/nemo_vfm/physicalai/tokenizer/data/augmentors/__init__.py b/nemo_dfm/physicalai/tokenizer/data/augmentors/__init__.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/data/augmentors/__init__.py rename to nemo_dfm/physicalai/tokenizer/data/augmentors/__init__.py diff --git a/nemo_vfm/physicalai/tokenizer/data/augmentors/augmentor.py b/nemo_dfm/physicalai/tokenizer/data/augmentors/augmentor.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/data/augmentors/augmentor.py rename to nemo_dfm/physicalai/tokenizer/data/augmentors/augmentor.py diff --git a/nemo_vfm/physicalai/tokenizer/data/augmentors/image/__init__.py b/nemo_dfm/physicalai/tokenizer/data/augmentors/image/__init__.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/data/augmentors/image/__init__.py rename to nemo_dfm/physicalai/tokenizer/data/augmentors/image/__init__.py diff --git a/nemo_vfm/physicalai/tokenizer/data/augmentors/image/cropping.py b/nemo_dfm/physicalai/tokenizer/data/augmentors/image/cropping.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/data/augmentors/image/cropping.py rename to nemo_dfm/physicalai/tokenizer/data/augmentors/image/cropping.py diff --git a/nemo_vfm/physicalai/tokenizer/data/augmentors/image/misc.py b/nemo_dfm/physicalai/tokenizer/data/augmentors/image/misc.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/data/augmentors/image/misc.py rename to nemo_dfm/physicalai/tokenizer/data/augmentors/image/misc.py diff --git a/nemo_vfm/physicalai/tokenizer/data/augmentors/image/normalize.py b/nemo_dfm/physicalai/tokenizer/data/augmentors/image/normalize.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/data/augmentors/image/normalize.py rename to nemo_dfm/physicalai/tokenizer/data/augmentors/image/normalize.py diff --git a/nemo_vfm/physicalai/tokenizer/data/augmentors/image/resize.py b/nemo_dfm/physicalai/tokenizer/data/augmentors/image/resize.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/data/augmentors/image/resize.py rename to nemo_dfm/physicalai/tokenizer/data/augmentors/image/resize.py diff --git a/nemo_vfm/physicalai/tokenizer/data/dataloader.py b/nemo_dfm/physicalai/tokenizer/data/dataloader.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/data/dataloader.py rename to nemo_dfm/physicalai/tokenizer/data/dataloader.py diff --git a/nemo_vfm/physicalai/tokenizer/data/utils.py b/nemo_dfm/physicalai/tokenizer/data/utils.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/data/utils.py rename to nemo_dfm/physicalai/tokenizer/data/utils.py diff --git a/nemo_vfm/physicalai/tokenizer/losses/__init__.py b/nemo_dfm/physicalai/tokenizer/losses/__init__.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/losses/__init__.py rename to nemo_dfm/physicalai/tokenizer/losses/__init__.py diff --git a/nemo_vfm/physicalai/tokenizer/losses/config.py b/nemo_dfm/physicalai/tokenizer/losses/config.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/losses/config.py rename to nemo_dfm/physicalai/tokenizer/losses/config.py diff --git a/nemo_vfm/physicalai/tokenizer/losses/loss.py b/nemo_dfm/physicalai/tokenizer/losses/loss.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/losses/loss.py rename to nemo_dfm/physicalai/tokenizer/losses/loss.py diff --git a/nemo_vfm/physicalai/tokenizer/losses/lpips.py b/nemo_dfm/physicalai/tokenizer/losses/lpips.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/losses/lpips.py rename to nemo_dfm/physicalai/tokenizer/losses/lpips.py diff --git a/nemo_vfm/physicalai/tokenizer/losses/test.py b/nemo_dfm/physicalai/tokenizer/losses/test.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/losses/test.py rename to nemo_dfm/physicalai/tokenizer/losses/test.py diff --git a/nemo_vfm/physicalai/tokenizer/test_tokenizer_model.py b/nemo_dfm/physicalai/tokenizer/test_tokenizer_model.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/test_tokenizer_model.py rename to nemo_dfm/physicalai/tokenizer/test_tokenizer_model.py diff --git a/nemo_vfm/physicalai/tokenizer/tokenizer_model.py b/nemo_dfm/physicalai/tokenizer/tokenizer_model.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/tokenizer_model.py rename to nemo_dfm/physicalai/tokenizer/tokenizer_model.py diff --git a/nemo_vfm/physicalai/tokenizer/train_tokenizer.py b/nemo_dfm/physicalai/tokenizer/train_tokenizer.py similarity index 100% rename from nemo_vfm/physicalai/tokenizer/train_tokenizer.py rename to nemo_dfm/physicalai/tokenizer/train_tokenizer.py diff --git a/nemo_vfm/sparse_attention/README.md b/nemo_dfm/sparse_attention/README.md similarity index 100% rename from nemo_vfm/sparse_attention/README.md rename to nemo_dfm/sparse_attention/README.md diff --git a/nemo_vfm/sparse_attention/accel-config.yml b/nemo_dfm/sparse_attention/accel-config.yml similarity index 100% rename from nemo_vfm/sparse_attention/accel-config.yml rename to nemo_dfm/sparse_attention/accel-config.yml diff --git a/nemo_vfm/sparse_attention/kernels/tl_base_attention.py b/nemo_dfm/sparse_attention/kernels/tl_base_attention.py similarity index 100% rename from nemo_vfm/sparse_attention/kernels/tl_base_attention.py rename to nemo_dfm/sparse_attention/kernels/tl_base_attention.py diff --git a/nemo_vfm/sparse_attention/kernels/tl_blocksum_attention.py b/nemo_dfm/sparse_attention/kernels/tl_blocksum_attention.py similarity index 100% rename from nemo_vfm/sparse_attention/kernels/tl_blocksum_attention.py rename to nemo_dfm/sparse_attention/kernels/tl_blocksum_attention.py diff --git a/nemo_vfm/sparse_attention/kernels/tl_sparse_attention.py b/nemo_dfm/sparse_attention/kernels/tl_sparse_attention.py similarity index 100% rename from nemo_vfm/sparse_attention/kernels/tl_sparse_attention.py rename to nemo_dfm/sparse_attention/kernels/tl_sparse_attention.py diff --git a/nemo_vfm/sparse_attention/sparse_attn.py b/nemo_dfm/sparse_attention/sparse_attn.py similarity index 100% rename from nemo_vfm/sparse_attention/sparse_attn.py rename to nemo_dfm/sparse_attention/sparse_attn.py diff --git a/nemo_vfm/sparse_attention/training.md b/nemo_dfm/sparse_attention/training.md similarity index 82% rename from nemo_vfm/sparse_attention/training.md rename to nemo_dfm/sparse_attention/training.md index a27f0ffc..53c33417 100644 --- a/nemo_vfm/sparse_attention/training.md +++ b/nemo_dfm/sparse_attention/training.md @@ -2,12 +2,12 @@ Clone the repository ```bash git clone https://gitlab-master.nvidia.com/sgovande/Cosmos/ Cosmos-gitlab -git clone https://gitlab-master.nvidia.com/dl/nemo/nemo-vfm nemo-vfm -cd nemo-vfm && git checkout training-sparse-attn && cd .. +git clone https://gitlab-master.nvidia.com/dl/nemo/nemo-vfm nemo-dfm +cd nemo-dfm && git checkout training-sparse-attn && cd .. ``` Modify `Cosmos-gitlab/cosmos1/models/diffusion/nemo/post_training/multinode.sh` with the following: -1. Update paths to `nemo-vfm` +1. Update paths to `nemo-dfm` 2. Update the environment variable `MOUNT` to point to your new location ```bash diff --git a/pyproject.toml b/pyproject.toml index 20a4d190..f1df75df 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -13,12 +13,12 @@ # limitations under the License. [project] -name = "nemo-vfm" +name = "nemo-dfm" dynamic = [ "version", "readme", ] -description = "NeMo VFM" +description = "NeMo DFM" license = {file = "LICENSE"} requires-python = ">=3.10" authors = [{ name = "NVIDIA", email = "nemo-toolkit@nvidia.com" }] @@ -84,15 +84,15 @@ dev = [ ] [tool.setuptools] -packages = ["nemo_vfm"] +packages = ["nemo_dfm"] [tool.setuptools.dynamic] -version = {attr = "nemo_vfm.__version__"} +version = {attr = "nemo_dfm.__version__"} readme = {file = "README.md", content-type = "text/markdown"} [project.urls] -Download = "https://github.com/NVIDIA-NeMo/NeMo-VFM/releases" -Homepage = "https://github.com/NVIDIA-NeMo/NeMo-VFM" +Download = "https://github.com/NVIDIA-NeMo/DFM/releases" +Homepage = "https://github.com/NVIDIA-NeMo/DFM" [tool.pytest.ini_options] # durations=0 will display all tests execution time, sorted in ascending order starting from from the slowest one. @@ -101,7 +101,7 @@ addopts = "--verbose --pyargs --durations=0 --strict-markers" # always add thes testpaths = ["tests"] # directories to ignore when discovering tests norecursedirs = [ - "nemo_vfm", + "nemo_dfm", "external", "examples", "docs", @@ -172,7 +172,7 @@ ignore = [ "D103", # Missing docstring in public function ] [tool.ruff.lint.isort] -known-first-party = ["nemo_vfm"] +known-first-party = ["nemo_dfm"] known-third-party = ["examples", "scripts"] section-order = ["future", "standard-library", "third-party", "first-party", "local-folder"] lines-after-imports = 2 @@ -196,4 +196,4 @@ concurrency = ["thread", "multiprocessing"] omit = ["/tmp/*"] [tool.coverage.paths] -source = ["nemo_vfm/", "/workspace/nemo_vfm"] +source = ["nemo_dfm/", "/workspace/nemo_dfm"]