make slicing vae compatible with latest diffusers

kohya-ss · Nov 26, 2023 · 764e333 · 764e333
1 parent c61e3bf
commit 764e333
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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