vqvae_advection_256 / config.json
sogeeking's picture
Upload tokenizer
d74ad14 verified
raw
history blame
975 Bytes
{
"_commit_hash": null,
"architectures": [
"Tokenizer"
],
"channels": 1,
"code_dim": 256,
"codebook_size": 256,
"disc_start": 5001,
"fsq_levels": [
8,
8,
6,
6,
6
],
"image_size": 256,
"init_dim": 128,
"kl_weight": 1e-05,
"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
}