Spaces:
Paused
Paused
Commit
•
67cb1f8
1
Parent(s):
7822350
Fix import issue
Browse files
SyncDreamer/configs/syncdreamer-train.yaml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
model:
|
| 2 |
base_learning_rate: 5.0e-05
|
| 3 |
-
target: ldm.models.diffusion.sync_dreamer.SyncMultiviewDiffusion
|
| 4 |
params:
|
| 5 |
view_num: 16
|
| 6 |
image_size: 256
|
|
@@ -10,10 +10,10 @@ model:
|
|
| 10 |
finetune_unet: false
|
| 11 |
finetune_projection: false
|
| 12 |
drop_conditions: false
|
| 13 |
-
clip_image_encoder_path: ckpt/ViT-L-14.pt
|
| 14 |
|
| 15 |
scheduler_config: # 10000 warmup steps
|
| 16 |
-
target: ldm.lr_scheduler.LambdaLinearScheduler
|
| 17 |
params:
|
| 18 |
warm_up_steps: [ 100 ]
|
| 19 |
cycle_lengths: [ 100000 ]
|
|
@@ -22,7 +22,7 @@ model:
|
|
| 22 |
f_min: [ 1.0 ]
|
| 23 |
|
| 24 |
unet_config:
|
| 25 |
-
target: ldm.models.diffusion.sync_dreamer_attention.DepthWiseAttention
|
| 26 |
params:
|
| 27 |
volume_dims: [64, 128, 256, 512]
|
| 28 |
image_size: 32
|
|
@@ -36,11 +36,11 @@ model:
|
|
| 36 |
use_spatial_transformer: True
|
| 37 |
transformer_depth: 1
|
| 38 |
context_dim: 768
|
| 39 |
-
use_checkpoint:
|
| 40 |
legacy: False
|
| 41 |
|
| 42 |
data:
|
| 43 |
-
target: ldm.data.sync_dreamer.SyncDreamerDataset
|
| 44 |
params:
|
| 45 |
target_dir: training_examples/target # renderings of target views
|
| 46 |
input_dir: training_examples/input # renderings of input views
|
|
|
|
| 1 |
model:
|
| 2 |
base_learning_rate: 5.0e-05
|
| 3 |
+
target: SyncDreamer.ldm.models.diffusion.sync_dreamer.SyncMultiviewDiffusion
|
| 4 |
params:
|
| 5 |
view_num: 16
|
| 6 |
image_size: 256
|
|
|
|
| 10 |
finetune_unet: false
|
| 11 |
finetune_projection: false
|
| 12 |
drop_conditions: false
|
| 13 |
+
clip_image_encoder_path: SyncDreamer/ckpt/ViT-L-14.pt
|
| 14 |
|
| 15 |
scheduler_config: # 10000 warmup steps
|
| 16 |
+
target: SyncDreamer.ldm.lr_scheduler.LambdaLinearScheduler
|
| 17 |
params:
|
| 18 |
warm_up_steps: [ 100 ]
|
| 19 |
cycle_lengths: [ 100000 ]
|
|
|
|
| 22 |
f_min: [ 1.0 ]
|
| 23 |
|
| 24 |
unet_config:
|
| 25 |
+
target: SyncDreamer.ldm.models.diffusion.sync_dreamer_attention.DepthWiseAttention
|
| 26 |
params:
|
| 27 |
volume_dims: [64, 128, 256, 512]
|
| 28 |
image_size: 32
|
|
|
|
| 36 |
use_spatial_transformer: True
|
| 37 |
transformer_depth: 1
|
| 38 |
context_dim: 768
|
| 39 |
+
use_checkpoint: False
|
| 40 |
legacy: False
|
| 41 |
|
| 42 |
data:
|
| 43 |
+
target: SyncDreamer.ldm.data.sync_dreamer.SyncDreamerDataset
|
| 44 |
params:
|
| 45 |
target_dir: training_examples/target # renderings of target views
|
| 46 |
input_dir: training_examples/input # renderings of input views
|
SyncDreamer/configs/syncdreamer.yaml
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
model:
|
| 2 |
base_learning_rate: 5.0e-05
|
| 3 |
-
target: ldm.models.diffusion.sync_dreamer.SyncMultiviewDiffusion
|
| 4 |
params:
|
| 5 |
view_num: 16
|
| 6 |
image_size: 256
|
|
@@ -10,10 +10,10 @@ model:
|
|
| 10 |
finetune_unet: false
|
| 11 |
finetune_projection: false
|
| 12 |
drop_conditions: false
|
| 13 |
-
clip_image_encoder_path: ckpt/ViT-L-14.pt
|
| 14 |
|
| 15 |
scheduler_config: # 10000 warmup steps
|
| 16 |
-
target: ldm.lr_scheduler.LambdaLinearScheduler
|
| 17 |
params:
|
| 18 |
warm_up_steps: [ 100 ]
|
| 19 |
cycle_lengths: [ 100000 ]
|
|
@@ -22,7 +22,7 @@ model:
|
|
| 22 |
f_min: [ 1.0 ]
|
| 23 |
|
| 24 |
unet_config:
|
| 25 |
-
target: ldm.models.diffusion.sync_dreamer_attention.DepthWiseAttention
|
| 26 |
params:
|
| 27 |
volume_dims: [64, 128, 256, 512]
|
| 28 |
image_size: 32
|
|
@@ -36,7 +36,7 @@ model:
|
|
| 36 |
use_spatial_transformer: True
|
| 37 |
transformer_depth: 1
|
| 38 |
context_dim: 768
|
| 39 |
-
use_checkpoint:
|
| 40 |
legacy: False
|
| 41 |
|
| 42 |
data: {}
|
|
|
|
| 1 |
model:
|
| 2 |
base_learning_rate: 5.0e-05
|
| 3 |
+
target: SyncDreamer.ldm.models.diffusion.sync_dreamer.SyncMultiviewDiffusion
|
| 4 |
params:
|
| 5 |
view_num: 16
|
| 6 |
image_size: 256
|
|
|
|
| 10 |
finetune_unet: false
|
| 11 |
finetune_projection: false
|
| 12 |
drop_conditions: false
|
| 13 |
+
clip_image_encoder_path: SyncDreamer/ckpt/ViT-L-14.pt
|
| 14 |
|
| 15 |
scheduler_config: # 10000 warmup steps
|
| 16 |
+
target: SyncDreamer.ldm.lr_scheduler.LambdaLinearScheduler
|
| 17 |
params:
|
| 18 |
warm_up_steps: [ 100 ]
|
| 19 |
cycle_lengths: [ 100000 ]
|
|
|
|
| 22 |
f_min: [ 1.0 ]
|
| 23 |
|
| 24 |
unet_config:
|
| 25 |
+
target: SyncDreamer.ldm.models.diffusion.sync_dreamer_attention.DepthWiseAttention
|
| 26 |
params:
|
| 27 |
volume_dims: [64, 128, 256, 512]
|
| 28 |
image_size: 32
|
|
|
|
| 36 |
use_spatial_transformer: True
|
| 37 |
transformer_depth: 1
|
| 38 |
context_dim: 768
|
| 39 |
+
use_checkpoint: False
|
| 40 |
legacy: False
|
| 41 |
|
| 42 |
data: {}
|