aicentreflip's picture
Publish the FLIP tutorial models
d4700c7 verified
Raw History Blame Contribute Delete
621 Bytes
# The frozen autoencoder in autoencoder.safetensors: the architecture every client builds, and the
# fingerprint of its weights, in the form process_tools/extract_autoencoder.py writes.
autoencoder:
spatial_dims: 3
in_channels: 1
out_channels: 1
num_res_blocks:
- 2
- 2
- 2
channels:
- 64
- 128
- 256
attention_levels:
- false
- false
- false
latent_channels: 3
norm_num_groups: 32
norm_eps: 1.0e-06
with_encoder_nonlocal_attn: false
with_decoder_nonlocal_attn: false
autoencoder_sha256: 8576bf403e1ca857d4c2fc125ff89d5a1a42a99a77627a3fa33c3369c839b6d9 # pragma: allowlist secret