default.yaml 402 Bytes
Newer Older
chenpangpang's avatar
chenpangpang committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
transformer_additional_kwargs:
  basic_block_type:                           "basic"
  after_norm:                                 false
  time_position_encoding:                     true

noise_scheduler_kwargs:
  beta_start:    0.00085
  beta_end:      0.03
  beta_schedule: "scaled_linear"
  steps_offset:  1
  prediction_type: "v_prediction"
  clip_sample: false

vae_kwargs:
  enable_magvit: true