vae_advection_16_v2 / config.json
sogeeking's picture
Upload tokenizer
cd8c929 verified
raw
history blame
No virus
975 Bytes
{
"_commit_hash": null,
"architectures": [
"Tokenizer"
],
"channels": 1,
"code_dim": 16,
"codebook_size": 256,
"disc_start": 5001,
"fsq_levels": [
8,
8,
6,
6,
6
],
"image_size": 256,
"init_dim": 128,
"kl_weight": 0.0001,
"layers": [
"residual",
"residual",
"residual",
"residual",
"compress_space",
"residual",
"residual",
"residual",
"compress_space",
"residual",
"residual",
"residual",
"compress_space",
"residual",
"residual",
"residual",
"compress_space",
"residual",
"residual",
"residual"
],
"model_type": "pdetokenizer",
"num_codebooks": 1,
"num_groups": 8,
"pad_mode": "circular",
"perceptual_loss_weight": 0,
"quantization_type": "vq",
"temporal_compression": false,
"torch_dtype": "float32",
"transformers_version": null,
"use_batch_norm": false,
"use_gan": false,
"use_revin": false,
"use_style": false
}