Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

quick fix

This commit is contained in:
Patrick von Platen 2022-07-13 10:28:46 +00:00
parent 5e12d5c691
commit c8c0c0e846
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_modeling_utils.py
View File

@ -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"),