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.

Allow LDM model to train on non-quantized encoded outputs of the VQ-VAE #473

@virginiafdez

Description

@virginiafdez

Function encode_stage_2_inputs of the VQ-VAE outputs the quantised latent representations. Since decode_stage_2_outputs quantises the input anyway, we could have a flag in encode_stage_2_inputs to allow for the user to train an LDM on non-quantised representations.

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions