File size: 269 Bytes
5325fcc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
# @package _global_
diffusion_unet:
hidden: 48
depth: 4
res_blocks: 1
norm_groups: 4
kernel: 8
stride: 4
growth: 4
max_channels: 10_000
dropout: 0.
emb_all_layers: true
bilstm: false
codec_dim: null
transformer: false
cross_attention: false |