Moved diffdec out

cosmos1/models/autoregressive/diffusion_decoder/network.py → ar_diffusion_decoder_network.py

@@ -20,8 +20,8 @@ from einops import rearrange
 from torch import nn
 from torchvision import transforms
 
-from blocks import PatchEmbed
-from general_dit import GeneralDIT
+from .blocks import PatchEmbed
+from .general_dit import GeneralDIT
 
 
 class DiffusionDecoderGeneralDIT(GeneralDIT):

cosmos1/models/autoregressive/diffusion_decoder/config/inference/cosmos_diffusiondecoder_7b.py → cosmos_diffusiondecoder_7b.py

@@ -15,9 +15,9 @@
 
 from hydra.core.config_store import ConfigStore
 
-from cosmos1.models.autoregressive.diffusion_decoder.network import DiffusionDecoderGeneralDIT
-from lazy_config_init import LazyCall as L
-from lazy_config_init import LazyDict
+from .ar_diffusion_decoder_network import DiffusionDecoderGeneralDIT
+from .lazy_config_init import LazyCall as L
+from .lazy_config_init import LazyDict
 
 num_frames = 57
 Cosmos_DiffusionDecoder_7B_INFERENCE_ONLY: LazyDict = LazyDict(