We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a560756 commit 7ca0676Copy full SHA for 7ca0676
monai/networks/nets/diffusion_model_unet.py
@@ -1891,7 +1891,7 @@ class DiffusionModelEncoder(nn.Module):
1891
norm_num_groups: number of groups for the normalization.
1892
norm_eps: epsilon for the normalization.
1893
resblock_updown: if True use residual blocks for downsampling.
1894
- num_head_channels: number of channels in each attention head.
+ num_head_channels: number of channels in each attention head.
1895
with_conditioning: if True add spatial transformers to perform conditioning.
1896
transformer_num_layers: number of layers of Transformer blocks to use.
1897
cross_attention_dim: number of context dimensions to use.
0 commit comments