Fix import issue

SyncDreamer/configs/syncdreamer-train.yaml

@@ -1,6 +1,6 @@
 model:
   base_learning_rate: 5.0e-05
-  target: ldm.models.diffusion.sync_dreamer.SyncMultiviewDiffusion
+  target: SyncDreamer.ldm.models.diffusion.sync_dreamer.SyncMultiviewDiffusion
   params:
     view_num: 16
     image_size: 256
@@ -10,10 +10,10 @@ model:
     finetune_unet: false
     finetune_projection: false
     drop_conditions: false
-    clip_image_encoder_path: ckpt/ViT-L-14.pt
+    clip_image_encoder_path: SyncDreamer/ckpt/ViT-L-14.pt
 
     scheduler_config: # 10000 warmup steps
-      target: ldm.lr_scheduler.LambdaLinearScheduler
+      target: SyncDreamer.ldm.lr_scheduler.LambdaLinearScheduler
       params:
         warm_up_steps: [ 100 ]
         cycle_lengths: [ 100000 ]
@@ -22,7 +22,7 @@ model:
         f_min: [ 1.0 ]
 
     unet_config:
-      target: ldm.models.diffusion.sync_dreamer_attention.DepthWiseAttention
+      target: SyncDreamer.ldm.models.diffusion.sync_dreamer_attention.DepthWiseAttention
       params:
         volume_dims: [64, 128, 256, 512]
         image_size: 32
@@ -36,11 +36,11 @@ model:
         use_spatial_transformer: True
         transformer_depth: 1
         context_dim: 768
-        use_checkpoint: True
+        use_checkpoint: False
         legacy: False
 
 data:
-  target: ldm.data.sync_dreamer.SyncDreamerDataset
+  target: SyncDreamer.ldm.data.sync_dreamer.SyncDreamerDataset
   params:
     target_dir: training_examples/target # renderings of target views
     input_dir: training_examples/input # renderings of input views

SyncDreamer/configs/syncdreamer.yaml

@@ -1,6 +1,6 @@
 model:
   base_learning_rate: 5.0e-05
-  target: ldm.models.diffusion.sync_dreamer.SyncMultiviewDiffusion
+  target: SyncDreamer.ldm.models.diffusion.sync_dreamer.SyncMultiviewDiffusion
   params:
     view_num: 16
     image_size: 256
@@ -10,10 +10,10 @@ model:
     finetune_unet: false
     finetune_projection: false
     drop_conditions: false
-    clip_image_encoder_path: ckpt/ViT-L-14.pt
+    clip_image_encoder_path: SyncDreamer/ckpt/ViT-L-14.pt
 
     scheduler_config: # 10000 warmup steps
-      target: ldm.lr_scheduler.LambdaLinearScheduler
+      target: SyncDreamer.ldm.lr_scheduler.LambdaLinearScheduler
       params:
         warm_up_steps: [ 100 ]
         cycle_lengths: [ 100000 ]
@@ -22,7 +22,7 @@ model:
         f_min: [ 1.0 ]
 
     unet_config:
-      target: ldm.models.diffusion.sync_dreamer_attention.DepthWiseAttention
+      target: SyncDreamer.ldm.models.diffusion.sync_dreamer_attention.DepthWiseAttention
       params:
         volume_dims: [64, 128, 256, 512]
         image_size: 32
@@ -36,7 +36,7 @@ model:
         use_spatial_transformer: True
         transformer_depth: 1
         context_dim: 768
-        use_checkpoint: True
+        use_checkpoint: False
         legacy: False
 
 data: {}