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 59325dc commit 2a75fe3Copy full SHA for 2a75fe3
monai/networks/nets/diffusion_model_unet.py
@@ -1894,7 +1894,7 @@ class DiffusionModelEncoder(nn.Module):
1894
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.
+ cross_attention_dim: number of context dimensions to use.
1898
num_class_embeds: if specified (as an int), then this model will be class-conditional with `num_class_embeds` classes.
1899
upcast_attention: if True, upcast attention operations to full precision.
1900
"""
0 commit comments