| 6 | |
| 7 | @dataclass |
| 8 | class AutoencoderConfig: |
| 9 | in_channels: int = 3 |
| 10 | out_channels: int = 3 |
| 11 | latent_channels_out: int = 8 |
| 12 | latent_channels_in: int = 4 |
| 13 | block_out_channels: Tuple[int] = (128, 256, 512, 512) |
| 14 | layers_per_block: int = 2 |
| 15 | norm_num_groups: int = 32 |
| 16 | scaling_factor: float = 0.18215 |
| 17 | |
| 18 | |
| 19 | @dataclass |