quick fix

c8c0c0e8 · Patrick von Platen · 5e12d5c6 · c8c0c0e8
Commit c8c0c0e8 authored Jul 13, 2022 by Patrick von Platen
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

tests/test_modeling_utils.py tests/test_modeling_utils.py +1 -1

No files found.
--- 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"),