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.
Remove autoencoder_model.train() from inferer tests #245
Description
Metadata
Metadata
Assignees
Labels
No labels
In the latent diffusion inferer tests, we are using an unnecessary
autoencoder_model.train()and not setting stage_2 to eval model. For example,GenerativeModels/tests/test_latent_diffusion_inferer.py
Line 94 in aef9ba2