2023-07-09 21:32:22 +08:00
|
|
|
unet_additional_kwargs:
|
|
|
|
|
use_motion_module: true
|
|
|
|
|
motion_module_resolutions:
|
|
|
|
|
- 1
|
|
|
|
|
- 2
|
|
|
|
|
- 4
|
|
|
|
|
- 8
|
|
|
|
|
motion_module_mid_block: 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-11-10 11:57:39 +08:00
|
|
|
temporal_position_encoding_max_len: 32
|
2023-07-09 21:32:22 +08:00
|
|
|
temporal_attention_dim_div: 1
|
|
|
|
|
|
2023-11-10 11:57:39 +08:00
|
|
|
|
2023-07-09 21:32:22 +08:00
|
|
|
noise_scheduler_kwargs:
|
2023-11-10 11:57:39 +08:00
|
|
|
num_train_timesteps: 1000
|
|
|
|
|
beta_start: 0.00085
|
|
|
|
|
beta_end: 0.020
|
|
|
|
|
beta_schedule: "scaled_linear"
|
|
|
|
|
|
|
|
|
|
|
