mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-01-13 23:08:00 -05: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
|