sound_tokenizer/config.json

{
  "_class_name": "Cosmos3AVAEAudioTokenizer",
  "_diffusers_version": "0.40.0.dev0",
  "_name_or_path": "/tmp/cosmos3-tiny-modular-prototype-20260715",
  "activation": "snakebeta",
  "anti_aliasing": false,
  "bottleneck": null,
  "bottleneck_type": "vae",
  "causal": false,
  "dec_anti_aliasing": false,
  "dec_c_mults": [
    1,
    2
  ],
  "dec_dim": 4,
  "dec_final_tanh": false,
  "dec_out_channels": 2,
  "dec_strides": [
    2,
    2
  ],
  "dec_type": "oobleck",
  "dec_use_nearest_upsample": false,
  "dec_use_snake": true,
  "dec_use_tanh_at_final": false,
  "enc_c_mults": [
    1
  ],
  "enc_dim": 4,
  "enc_hop_length": 2,
  "enc_identity_init": false,
  "enc_intermediate_dim": 768,
  "enc_latent_dim": 8,
  "enc_n_fft": 8,
  "enc_num_blocks": 1,
  "enc_num_layers": 12,
  "enc_strides": [
    2
  ],
  "enc_type": "spec_convnext",
  "enc_use_snake": true,
  "encoder_enabled": true,
  "hop_size": 4,
  "input_channels": 1,
  "latent_mean": null,
  "latent_std": null,
  "model_type": "autoencoder_v2",
  "normalize_volume": false,
  "padding_mode": "zeros",
  "sampling_rate": 16,
  "snake_logscale": true,
  "stereo": true,
  "use_cuda_kernel": false,
  "use_wav_as_input": true,
  "vocoder_input_dim": 4
}