From 764e333fa2b560f2c5584f5ce9fb351467a7a218 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Sun, 26 Nov 2023 18:12:04 +0900 Subject: [PATCH] make slicing vae compatible with latest diffusers --- library/slicing_vae.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/library/slicing_vae.py b/library/slicing_vae.py index 31b2bd0a..5c4e056d 100644 --- a/library/slicing_vae.py +++ b/library/slicing_vae.py @@ -62,7 +62,7 @@ def cat_h(sliced): return x -def resblock_forward(_self, num_slices, input_tensor, temb): +def resblock_forward(_self, num_slices, input_tensor, temb, **kwargs): assert _self.upsample is None and _self.downsample is None assert _self.norm1.num_groups == _self.norm2.num_groups assert temb is None