sparse1024/vae/config.json

{
  "model_type": "vae",
  "model_class": "pixal3d.models.autoencoders.ss_vae.SparseSDFVAE",
  "use_checkpoint": true,
  "embed_dim": 16,
  "num_head_channels_encoder": 64,
  "model_channels_encoder": 512,
  "num_heads_encoder": 8,
  "num_blocks_encoder": 4,
  "num_head_channels_decoder": 64,
  "model_channels_decoder": 512,
  "num_heads_decoder": 8,
  "num_blocks_decoder": 4,
  "resolution": 128,
  "out_channels": 1,
  "use_fp16": true,
  "latents_scale": 1.0,
  "latents_shift": 0.0,
  "chunk_size": 4
}