benjamin-paine's picture
Add Diffusers AutoModel support (#1)
9f80e4d verified
raw
history blame contribute delete
643 Bytes
{
"_class_name": "Flux2TinyAutoEncoder",
"_diffusers_version": "0.35.2",
"auto_map": {
"AutoModel": "flux2_tiny_autoencoder.Flux2TinyAutoEncoder"
},
"act_fn": "silu",
"decoder_block_out_channels": [
64,
64,
64,
64
],
"encoder_block_out_channels": [
64,
64,
64,
64
],
"force_upcast": false,
"in_channels": 3,
"latent_channels": 128,
"latent_magnitude": 3.0,
"latent_shift": 0.5,
"num_decoder_blocks": [
3,
3,
3,
1
],
"num_encoder_blocks": [
1,
3,
3,
3
],
"out_channels": 3,
"scaling_factor": 0.13025,
"upsampling_scaling_factor": 2
}