2023-07-09 21:32:22 +08:00
|
|
|
unet_additional_kwargs:
|
2023-12-15 20:57:51 +08:00
|
|
|
use_inflated_groupnorm: true
|
2023-07-09 21:32:22 +08:00
|
|
|
unet_use_cross_frame_attention: false
|
2023-12-15 20:57:51 +08:00
|
|
|
unet_use_temporal_attention: false
|
|
|
|
|
use_motion_module: true
|
|
|
|
|
motion_module_resolutions: [1,2,4,8]
|
|
|
|
|
motion_module_mid_block: true
|
|
|
|
|
motion_module_decoder_only: false
|
|
|
|
|
motion_module_type: "Vanilla"
|
|
|
|
|
|
2023-07-09 21:32:22 +08:00
|
|
|
motion_module_kwargs:
|
2023-12-15 20:57:51 +08:00
|
|
|
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-12-15 20:57:51 +08:00
|
|
|
temporal_attention_dim_div: 1
|
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
|
|
|
|
|
clip_sample: False
|
