File size: 500 Bytes
8a37e0a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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
|