Project-MONAI / GenerativeModels

MONAI Generative Models makes it easy to train, evaluate, and deploy generative models and related applications
Apache License 2.0
555 stars 78 forks source link

Allowed for the quantized flag to be passed to the LatentDiffusionInf… #481

Closed virginiafdez closed 3 months ago

virginiafdez commented 3 months ago

…erer methods, which is then passed to VQVAE encode_stage_2_inputs if autoencoder_model is a VQVAE.

Set this flag randomly during testing (when the autoencoder is a VAE, it shouldn't matter), ran the tests, and ran reformatting.