mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
14 lines
500 B
Python
14 lines
500 B
Python
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
|