From c8c0c0e846c8afc07602c44180278a2f7f15331d Mon Sep 17 00:00:00 2001 From: Patrick von Platen Date: Wed, 13 Jul 2022 10:28:46 +0000 Subject: [PATCH] quick fix --- tests/test_modeling_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_modeling_utils.py b/tests/test_modeling_utils.py index 2df3b21a..7cb640c9 100755 --- a/tests/test_modeling_utils.py +++ b/tests/test_modeling_utils.py @@ -493,7 +493,7 @@ class UNetLDMModelTests(ModelTesterMixin, unittest.TestCase): "num_res_blocks": 2, "attention_resolutions": (16,), "block_input_channels": [32, 32], - "resnet_output_channels": [32, 64], + "block_output_channels": [32, 64], "num_head_channels": 32, "conv_resample": True, "down_blocks": ("UNetResDownBlock2D", "UNetResDownBlock2D"),