put _hf inside, modify config correspondingly

config.json

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

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

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