File size: 685 Bytes
d0ffe9c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
{
"unet_additional_kwargs": {
"unet_use_temporal_attention": false,
"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,
"temporal_position_encoding_max_len": 32,
"temporal_attention_dim_div": 1
}
},
"noise_scheduler_kwargs": {
"num_train_timesteps": 1000,
"beta_start": 0.00085,
"beta_end": 0.020,
"beta_schedule": "scaled_linear"
}
}
|