| from diffusers.models.autoencoders.autoencoder_kl import AutoencoderKL | |
| from invokeai.backend.stable_diffusion.vae_tiling import patch_vae_tiling_params | |
| def test_patch_vae_tiling_params(): | |
| """Smoke test the patch_vae_tiling_params(...) context manager. The main purpose of this unit test is to detect if | |
| diffusers ever changes the attributes of the AutoencoderKL class that we expect to exist. | |
| """ | |
| vae = AutoencoderKL() | |
| with patch_vae_tiling_params(vae, 1, 2, 3): | |
| pass | |