File size: 390 Bytes
10e809f
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
name: flexTokenizer
embedding_dim: 16
hidden_channels: 128
channel_multipliers:
- 1
- 2
- 2
- 4
encoder_layer_configs:
- 2
- 2
- 2
- 2
- 2
decoder_layer_configs:
- 2
- 2
- 2
- 2
- 2
use_adaptive_norm: true
use_learnable_up_down_sample: true
quantizer_config:
  quantize_type: vq
  embed_dim: 16
  num_embed: 8192
  commitment_loss_weight: 0.25
  use_l2_norm: true
  use_uniform_init: false