From 19e0c9ef5d9c16395269ffe45ff01403b015a8b2 Mon Sep 17 00:00:00 2001 From: Kijai <40791699+kijai@users.noreply.github.com> Date: Tue, 26 Mar 2024 15:31:10 +0200 Subject: [PATCH] Fix loading 512 non-interolation model --- configs/dynamicrafter_512_v1.yaml | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/configs/dynamicrafter_512_v1.yaml b/configs/dynamicrafter_512_v1.yaml index 104c50b..581e274 100644 --- a/configs/dynamicrafter_512_v1.yaml +++ b/configs/dynamicrafter_512_v1.yaml @@ -1,5 +1,5 @@ model: - target: lvdm.models.ddpm3d.LatentVisualDiffusion + target: .lvdm.models.ddpm3d.LatentVisualDiffusion params: rescale_betas_zero_snr: True parameterization: "v" @@ -22,7 +22,7 @@ model: fps_condition_type: 'fps' perframe_ae: True unet_config: - target: lvdm.modules.networks.openaimodel3d.UNetModel + target: .lvdm.modules.networks.openaimodel3d.UNetModel params: in_channels: 8 out_channels: 4 @@ -55,7 +55,7 @@ model: fs_condition: true first_stage_config: - target: lvdm.models.autoencoder.AutoencoderKL + target: .lvdm.models.autoencoder.AutoencoderKL params: embed_dim: 4 monitor: val/rec_loss @@ -78,18 +78,18 @@ model: target: torch.nn.Identity cond_stage_config: - target: lvdm.modules.encoders.condition.FrozenOpenCLIPEmbedder + target: .lvdm.modules.encoders.condition.FrozenOpenCLIPEmbedder params: freeze: true layer: "penultimate" img_cond_stage_config: - target: lvdm.modules.encoders.condition.FrozenOpenCLIPImageEmbedderV2 + target: .lvdm.modules.encoders.condition.FrozenOpenCLIPImageEmbedderV2 params: freeze: true image_proj_stage_config: - target: lvdm.modules.encoders.resampler.Resampler + target: .lvdm.modules.encoders.resampler.Resampler params: dim: 1024 depth: 4