parent
aac76a1fb0
commit
b7f0bc74a7
|
@ -26,7 +26,7 @@ class MaskDecoder(nn.Module):
|
|||
) -> None:
|
||||
"""
|
||||
Predicts masks given an image and prompt embeddings, using a
|
||||
tranformer architecture.
|
||||
transformer architecture.
|
||||
|
||||
Arguments:
|
||||
transformer_dim (int): the channel dimension of the transformer
|
||||
|
|
Loading…
Reference in New Issue