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