support args, add comments

config_helper.py

@@ -164,7 +164,7 @@ def import_all_modules_from_package(package_path: str, reload: bool = False, ski
         reload (bool): Flag to determine whether to reload modules if they're already imported.
         skip_underscore (bool): If True, skips importing modules that start with an underscore.
     """
-    return  # TODO: we do not use this
+    return  # we do not use this function
     log.debug(f"{'Reloading' if reload else 'Importing'} all modules from package {package_path}")
     package = importlib.import_module(package_path)
     package_directory = package.__path__

df_config_config.py

@@ -22,7 +22,7 @@ from .df_config_registry import register_configs
 from .config import Config as ori_Config
 from .config_helper import import_all_modules_from_package
 
-# I added importing here 
+# import config here, not use importlib 
 from .cosmos1diffusiontext2world import LazyDict
 from .cosmos1diffusionvideo2world import LazyDict
 

inference_utils.py

@@ -275,7 +275,7 @@ def load_model_by_config(
     model_class=DiffusionT2WModel,
 ):
     # TODO: We need to modify this for huggingface because the config file path is different
-    # config_module = get_config_module(config_file)
+    # config_module = get_config_module(config_file) # cosmos1/models/diffusion/config/config.py
     # config = importlib.import_module(config_module).make_config()
     if model_class in (DiffusionT2WModel, DiffusionV2WModel):
         config = make_config()

text2world_hf.py

@@ -130,5 +130,8 @@ class DiffusionText2World(PreTrainedModel):
     def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):
         # TODO: this only builds the class, but not download the checkpoints.
         config = kwargs["config"]
+        other_args = kwargs.copy()
+        other_args.pop("config")
+        config.update(other_args)
         model = cls(config)
         return model