modify import to .cosmos1

text2world_hf.py

@@ -3,12 +3,11 @@ import argparse
 import torch
 from transformers import PreTrainedModel, PretrainedConfig
 
-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
+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
 
 class DiffusionText2WorldConfig(PretrainedConfig):
     model_type = "DiffusionText2World" 

text2world_hf_helper.py

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