envelope _hf with a helper, try to solve import error

config.json

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

text2world_hf.py

@@ -3,11 +3,12 @@ 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 cosmos1.models.diffusion.inference.inference_utils import add_common_arguments
+from cosmos1.models.diffusion.inference.inference_utils import 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
 
 class DiffusionText2WorldConfig(PretrainedConfig):
     model_type = "DiffusionText2World" 

text2world_hf_helper.py

@@ -0,0 +1 @@
+from text2world_hf import DiffusionText2World, DiffusionText2WorldConfig