2023-07-09 21:32:22 +08:00
|
|
|
unet_additional_kwargs:
|
2023-09-10 21:27:27 +08:00
|
|
|
use_inflated_groupnorm: true
|
2023-07-09 21:32:22 +08:00
|
|
|
unet_use_cross_frame_attention: false
|
|
|
|
|
unet_use_temporal_attention: false
|
|
|
|
|
use_motion_module: true
|
|
|
|
|
motion_module_resolutions:
|
|
|
|
|
- 1
|
|
|
|
|
- 2
|
|
|
|
|
- 4
|
|
|
|
|
- 8
|
2023-09-10 21:27:27 +08:00
|
|
|
motion_module_mid_block: true
|
2023-07-09 21:32:22 +08:00
|
|
|
motion_module_decoder_only: false
|
|
|
|
|
motion_module_type: Vanilla
|
|
|
|
|
motion_module_kwargs:
|
|
|
|
|
num_attention_heads: 8
|
|
|
|
|
num_transformer_block: 1
|
|
|
|
|
attention_block_types:
|
|
|
|
|
- Temporal_Self
|
|
|
|
|
- Temporal_Self
|
|
|
|
|
temporal_position_encoding: true
|
2023-09-10 21:27:27 +08:00
|
|
|
temporal_position_encoding_max_len: 32
|
2023-07-09 21:32:22 +08:00
|
|
|
temporal_attention_dim_div: 1
|
|
|
|
|
|
|
|
|
|
noise_scheduler_kwargs:
|
|
|
|
|
beta_start: 0.00085
|
|
|
|
|
beta_end: 0.012
|
|
|
|
|
beta_schedule: "linear"
|
