mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
Add unit test for patch_vae_tiling_params().
This commit is contained in:
parent
79640ba14e
commit
a1b7dbfa54
13
tests/backend/stable_diffusion/test_vae_tiling.py
Normal file
13
tests/backend/stable_diffusion/test_vae_tiling.py
Normal file
@ -0,0 +1,13 @@
|
||||
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
|
Loading…
Reference in New Issue
Block a user