Merge pull request #37 from eltociear/patch-1

fix typo in mask_decoder.py
This commit is contained in:
ericmintun 2023-04-10 08:44:41 -07:00 committed by GitHub
commit f3d37bde02
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -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