inference-v2-checkpoint.yaml 661 B

123456789101112131415161718192021222324252627
  1. unet_additional_kwargs:
  2. use_inflated_groupnorm: true
  3. unet_use_cross_frame_attention: false
  4. unet_use_temporal_attention: false
  5. use_motion_module: true
  6. motion_module_resolutions:
  7. - 1
  8. - 2
  9. - 4
  10. - 8
  11. motion_module_mid_block: true
  12. motion_module_decoder_only: false
  13. motion_module_type: Vanilla
  14. motion_module_kwargs:
  15. num_attention_heads: 8
  16. num_transformer_block: 1
  17. attention_block_types:
  18. - Temporal_Self
  19. - Temporal_Self
  20. temporal_position_encoding: true
  21. temporal_position_encoding_max_len: 32
  22. temporal_attention_dim_div: 1
  23. noise_scheduler_kwargs:
  24. beta_start: 0.00085
  25. beta_end: 0.012
  26. beta_schedule: "linear"