Fix typo in unet_blocks.py (huggingface#353)

da03 · web-flow · commit 6c0ca5efa636 · 2022-09-04T18:01:14.000+02:00
Update unet_blocks.py

fix typo
diff --git a/src/diffusers/models/unet_blocks.py b/src/diffusers/models/unet_blocks.py
@@ -60,7 +60,7 @@ def get_down_block(
         )
     elif down_block_type == "CrossAttnDownBlock2D":
         if cross_attention_dim is None:
-            raise ValueError("cross_attention_dim must be specified for CrossAttnUpBlock2D")
+            raise ValueError("cross_attention_dim must be specified for CrossAttnDownBlock2D")
         return CrossAttnDownBlock2D(
             num_layers=num_layers,
             in_channels=in_channels,

Original file line number	Diff line number	Diff line change
`@@ -60,7 +60,7 @@ def get_down_block(`
`60`	`60`	`)`
`61`	`61`	`elif down_block_type == "CrossAttnDownBlock2D":`
`62`	`62`	`if cross_attention_dim is None:`
`63`		`- raise ValueError("cross_attention_dim must be specified for CrossAttnUpBlock2D")`
	`63`	`+ raise ValueError("cross_attention_dim must be specified for CrossAttnDownBlock2D")`
`64`	`64`	`return CrossAttnDownBlock2D(`
`65`	`65`	`num_layers=num_layers,`
`66`	`66`	`in_channels=in_channels,`