Skip to content

Commit e4e13f5

Browse files
author
Mark-ZhouWX
committed
do not apply projection for blip2
1 parent dc74199 commit e4e13f5

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

official/cv/segment-anything/segment_anything/modeling/prompt_encoder.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,10 @@ def __init__(
5656
self.no_mask_embed = nn.Embedding(1, embed_dim)
5757

5858
self.text_embeddings = nn.Embedding(1, embed_dim)
59-
self.text_proj = nn.Dense(in_channels=text_feature_dim, out_channels=embed_dim)
59+
# self.text_proj = nn.Dense(in_channels=text_feature_dim, out_channels=embed_dim)
60+
61+
self.text_proj = nn.Dense(in_channels=text_feature_dim, out_channels=embed_dim) \
62+
if text_feature_dim != embed_dim else nn.Identity() # clip need projection while blip2 not
6063

6164
def get_dense_pe(self) -> ms.Tensor:
6265
"""

0 commit comments

Comments
 (0)