alex
commited on
Commit
·
ad9b2a4
1
Parent(s):
ae726e0
redo
Browse files
sam2/modeling/sam/transformer.py
CHANGED
|
@@ -279,6 +279,8 @@ class Attention(nn.Module):
|
|
| 279 |
|
| 280 |
out = self._recombine_heads(out)
|
| 281 |
|
|
|
|
|
|
|
| 282 |
return out
|
| 283 |
|
| 284 |
|
|
|
|
| 279 |
|
| 280 |
out = self._recombine_heads(out)
|
| 281 |
|
| 282 |
+
out = self.out_proj(out)
|
| 283 |
+
|
| 284 |
return out
|
| 285 |
|
| 286 |
|