taef1 / config.json
silveroxides's picture
Update config.json
e1cd86a verified
raw
history blame
598 Bytes
{
"_class_name": "AutoencoderTiny",
"_diffusers_version": "0.30.0",
"act_fn": "relu",
"decoder_block_out_channels": [
64,
64,
64,
64
],
"encoder_block_out_channels": [
64,
64,
64,
64
],
"force_upcast": false,
"in_channels": 3,
"latent_channels": 16,
"latent_magnitude": 3,
"latent_shift": 0.5,
"num_decoder_blocks": [
3,
3,
3,
1
],
"num_encoder_blocks": [
1,
3,
3,
3
],
"out_channels": 3,
"scaling_factor": 1.0,
"shift_factor": 0.0,
"upsample_fn": "nearest",
"upsampling_scaling_factor": 2
}