File size: 443 Bytes
ed7d94b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
name: flexTokenizer
embedding_dim: 16
hidden_channels: 128
channel_multipliers:
- 1
- 1
- 2
- 2
- 4
encoder_layer_configs:
- 2
- 2
- 2
- 2
- 2
- 2
decoder_layer_configs:
- 2
- 2
- 2
- 2
- 2
- 2
use_adaptive_norm: true
use_learnable_up_down_sample: true
quantizer_config:
  quantize_type: gvq
  embed_dim: 16
  num_embed: 262144
  commitment_loss_weight: 0.25
  use_l2_norm: false
  use_uniform_init: true
  num_group: 16
  force_1d_clip: true