fix typo in mask_decoder.py
tranformer -> transformer
This commit is contained in:
committed by
GitHub
parent
aac76a1fb0
commit
b7f0bc74a7
@@ -26,7 +26,7 @@ class MaskDecoder(nn.Module):
|
|||||||
) -> None:
|
) -> None:
|
||||||
"""
|
"""
|
||||||
Predicts masks given an image and prompt embeddings, using a
|
Predicts masks given an image and prompt embeddings, using a
|
||||||
tranformer architecture.
|
transformer architecture.
|
||||||
|
|
||||||
Arguments:
|
Arguments:
|
||||||
transformer_dim (int): the channel dimension of the transformer
|
transformer_dim (int): the channel dimension of the transformer
|
||||||
|
|||||||
Reference in New Issue
Block a user