add relative import

cosmos1diffusiontext2world.py

@@ -15,7 +15,7 @@
 
 from hydra.core.config_store import ConfigStore
 
-from lazy_config_init import LazyDict
+from .lazy_config_init import LazyDict
 
 Cosmos_1_0_Diffusion_Text2World_7B: LazyDict = LazyDict(
     dict(

cosmos1diffusionvideo2world.py

@@ -15,9 +15,9 @@
 
 from hydra.core.config_store import ConfigStore
 
-from cosmos1.models.diffusion.networks.general_dit_video_conditioned import VideoExtendGeneralDIT
-from lazy_config_init import LazyCall as L
-from lazy_config_init import LazyDict
+from .general_dit_video_conditioned import VideoExtendGeneralDIT
+from .lazy_config_init import LazyCall as L
+from .lazy_config_init import LazyDict
 
 Cosmos_1_0_Diffusion_Video2World_7B: LazyDict = LazyDict(
     dict(

cosmos1/models/diffusion/networks/general_dit_video_conditioned.py → general_dit_video_conditioned.py

@@ -19,9 +19,9 @@ import torch
 from einops import rearrange
 from torch import nn
 
-from conditioner import DataType
-from blocks import TimestepEmbedding, Timesteps
-from general_dit import GeneralDIT
+from .conditioner import DataType
+from .blocks import TimestepEmbedding, Timesteps
+from .general_dit import GeneralDIT
 from .log import log