vqvae_burgers_norevin / config.json
sogeeking's picture
Upload tokenizer
1a22b82
raw
history blame contribute delete
No virus
839 Bytes
{
"_commit_hash": null,
"architectures": [
"Tokenizer"
],
"attn_dim_head": 32,
"channels": 1,
"codebook_size": 1024,
"fsq_levels": [
7,
5,
5,
5,
5
],
"image_size": 256,
"init_dim": 64,
"layers": [
"residual",
"residual",
"compress_space",
"residual",
"residual",
"compress_space",
"residual",
"residual",
"compress_space",
"residual",
"residual",
"compress_space",
"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
}