Fixed config loading

.gitignore

@@ -1 +1,2 @@
-huggingface.txt
+huggingface.txt
+checkpoints/

cosmos1/models/autoregressive/diffusion_decoder/config/config_latent_diffusion_decoder.py → config_latent_diffusion_decoder.py

@@ -17,10 +17,10 @@ from typing import Any, List
 
 import attrs
 
-from cosmos1.models.autoregressive.diffusion_decoder.config.registry import register_configs as register_dd_configs
+from cosmos1.models.autoregressive.diffusion_decoder.config.ar_diffusion_decoder_config_registry import register_configs as register_dd_configs
 from df_base_model import LatentDiffusionDecoderModelConfig
 from df_config_registry import register_configs
-from . import config
+from .cosmos1.models.autoregressive.diffusion_decoder.config import config
 from config_helper import import_all_modules_from_package
 
 

inference_utils.py

@@ -26,10 +26,12 @@ import torchvision.transforms.functional as transforms_F
 
 from .model_t2w import DiffusionT2WModel
 from .model_v2w import DiffusionV2WModel
+from .ar_diffusion_decoder_model import LatentDiffusionDecoderModel
 from .config_helper import get_config_module, override
 from .utils_io import load_from_fileobj
 from .misc import misc
 from .df_config_config import make_config
+from .ldd_config import make_config as ldd_make_config
 from .log import log
 
 TORCH_VERSION: Tuple[int, ...] = tuple(int(x) for x in torch.__version__.split(".")[:2])
@@ -279,8 +281,9 @@ def load_model_by_config(
     # config = importlib.import_module(config_module).make_config()
     if model_class in (DiffusionT2WModel, DiffusionV2WModel):
         config = make_config()
-    else:
-        raise NotImplementedError("TODO: didn't implement autoregression")
+    elif model_class in (LatentDiffusionDecoderModel):
+        config = ldd_make_config()
+        
 
     config = override(config, ["--", f"experiment={config_job_name}"])
 

ldd_config.py

@@ -0,0 +1,61 @@
+# 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.
+
+from typing import Any, List
+
+import attrs
+
+from .ar_diffusion_decoder_config_registry import register_configs as register_dd_configs
+from .df_base_model import LatentDiffusionDecoderModelConfig
+from .df_config_registry import register_configs
+from .config import Config as ori_Config
+from .config_helper import import_all_modules_from_package
+
+
+@attrs.define(slots=False)
+class Config(ori_Config):
+    # default config groups that will be used unless overwritten
+    # see config groups in registry.py
+    defaults: List[Any] = attrs.field(
+        factory=lambda: [
+            "_self_",
+            {"net": None},
+            {"conditioner": "basic"},
+            {"tokenizer": "tokenizer"},
+            {"tokenizer_corruptor": None},
+            {"latent_corruptor": None},
+            {"pixel_corruptor": None},
+            {"experiment": None},
+        ]
+    )
+
+
+def make_config():
+    c = Config(model=LatentDiffusionDecoderModelConfig())
+
+    # Specifying values through instances of attrs
+    c.job.project = "cosmos_video4"
+    c.job.group = "debug"
+    c.job.name = "delete_${now:%Y-%m-%d}_${now:%H-%M-%S}"
+
+    # Call this function to register config groups for advanced overriding.
+    register_configs()
+    register_dd_configs()
+
+    # experiment config are defined in the experiment folder
+    # call import_all_modules_from_package to register them
+    import_all_modules_from_package("cosmos1.models.diffusion.config.inference", reload=True)
+    import_all_modules_from_package("cosmos1.models.autoregressive.diffusion_decoder.config.inference", reload=True)
+    return c