2023-07-09 21:32:22 +08:00
|
|
|
unet_additional_kwargs:
|
2024-07-17 08:08:10 +00:00
|
|
|
use_inflated_groupnorm: false
|
|
|
|
|
use_motion_module: true
|
|
|
|
|
motion_module_resolutions: [1,2,4,8]
|
|
|
|
|
motion_module_mid_block: false
|
|
|
|
|
motion_module_type: "Vanilla"
|
2023-12-15 20:57:51 +08:00
|
|
|
|
2023-07-09 21:32:22 +08:00
|
|
|
motion_module_kwargs:
|
2024-07-17 08:08:10 +00:00
|
|
|
num_attention_heads: 8
|
|
|
|
|
num_transformer_block: 1
|
|
|
|
|
attention_block_types: [ "Temporal_Self", "Temporal_Self" ]
|
|
|
|
|
temporal_position_encoding: true
|
|
|
|
|
temporal_attention_dim_div: 1
|
|
|
|
|
zero_initialize: true
|
2023-07-09 21:32:22 +08:00
|
|
|
|
|
|
|
|
noise_scheduler_kwargs:
|
2023-12-15 20:57:51 +08:00
|
|
|
beta_start: 0.00085
|
|
|
|
|
beta_end: 0.012
|
2023-07-09 21:32:22 +08:00
|
|
|
beta_schedule: "linear"
|
2023-12-15 20:57:51 +08:00
|
|
|
steps_offset: 1
|
2024-07-17 08:08:10 +00:00
|
|
|
clip_sample: false
|
