Skip to content
This repository was archived by the owner on Feb 7, 2025. It is now read-only.
This repository was archived by the owner on Feb 7, 2025. It is now read-only.

Change ch_mult to num_channels in Autoencoder #216

@virginiafdez

Description

@virginiafdez

In AutoencoderKL, the encoder and decoder are created using ch_mult to calculate the number of channels. It'll be a good idea to use an input argument num_channels instead, which is the one followed by the LDM.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Relationships

None yet

Development

No branches or pull requests

Issue actions