File size: 889 Bytes
cfb7702
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
target: sgm.models.autoencoder.AutoencodingEngine
params:
  loss_config:
    target: torch.nn.Identity
  regularizer_config:
    target: sgm.modules.autoencoding.regularizers.DiagonalGaussianRegularizer
  encoder_config: 
    target: sgm.modules.diffusionmodules.model.Encoder
    params:
      attn_type: vanilla
      double_z: True
      z_channels: 4
      resolution: 256
      in_channels: 3
      out_ch: 3
      ch: 128
      ch_mult: [1, 2, 4, 4]
      num_res_blocks: 2
      attn_resolutions: []
      dropout: 0.0
  decoder_config:
    target: sgm.modules.autoencoding.temporal_ae.VideoDecoder
    params:
      attn_type: vanilla
      double_z: True
      z_channels: 4
      resolution: 256
      in_channels: 3
      out_ch: 3
      ch: 128
      ch_mult: [1, 2, 4, 4]
      num_res_blocks: 2
      attn_resolutions: []
      dropout: 0.0
      video_kernel_size: [3, 1, 1]