Update config_identity.yaml

config_identity.yaml

@@ -1,6 +1,6 @@
 model:
   base_learning_rate: 2.0e-05
-  target: latent_diffusion.ldm.models.diffusion.ddpm.LatentDiffusion
+  target: ldm.models.diffusion.ddpm.LatentDiffusion
   params:
     linear_start: 0.0015
     linear_end: 0.0195
@@ -18,7 +18,7 @@ model:
     monitor: val/loss_simple_ema
 
     unet_config:
-      target: latent_diffusion.ldm.modules.diffusionmodules.openaimodel.UNetModel
+      target: ldm.modules.diffusionmodules.openaimodel.UNetModel
       params:
         image_size: 256
         in_channels: 24
@@ -40,13 +40,13 @@ model:
         context_dim: 768
 
     first_stage_config:
-      target: latent_diffusion.ldm.models.autoencoder.IdentityFirstStage
+      target: ldm.models.autoencoder.IdentityFirstStage
       params:
         embed_dim: 3
         n_embed: 8192
 
     cond_stage_config:
-      target: latent_diffusion.ldm.modules.encoders.modules.GPTEmbedder
+      target: ldm.modules.encoders.modules.GPTEmbedder
       params:
         n_embed: 768
         n_layer: 12