move _hf.py to root

config.json

@@ -3,8 +3,8 @@
       "DiffusionText2World"
     ],
     "auto_map": {
-      "AutoConfig": "cosmos1.models.diffusion.inference.text2world_hf.DiffusionText2WorldConfig",
-      "AutoModel": "cosmos1.models.diffusion.inference.text2world_hf.DiffusionText2World"
+      "AutoConfig": "text2world_hf.DiffusionText2WorldConfig",
+      "AutoModel": "text2world_hf.DiffusionText2World"
     },
     "model_type": "AutoModel"
 }

cosmos1/models/diffusion/inference/text2world_hf.py → text2world_hf.py

@@ -3,10 +3,10 @@ import argparse
 import torch
 from transformers import PreTrainedModel, PretrainedConfig
 
-from .inference_utils import add_common_arguments, validate_args
-from .world_generation_pipeline import DiffusionText2WorldGenerationPipeline
-from ....utils import log, misc
-from ....utils.io import read_prompts_from_file, save_video
+from .cosmos1.models.diffusion.inference.inference_utils import add_common_arguments, validate_args
+from .cosmos1.models.diffusion.inference.world_generation_pipeline import DiffusionText2WorldGenerationPipeline
+from .cosmos1.utils import log, misc
+from .cosmos1.utils.io import read_prompts_from_file, save_video
 
 class DiffusionText2WorldConfig(PretrainedConfig):
     model_type = "DiffusionText2World"