Update modeling_videollama3_encoder.py
Browse files
modeling_videollama3_encoder.py
CHANGED
@@ -343,7 +343,7 @@ class VisionSdpaAttention(VisionAttention):
|
|
343 |
attn_output = F.scaled_dot_product_attention(query_states, key_states, value_states, attention_mask, dropout_p=0.0)
|
344 |
attn_output = attn_output.transpose(0, 1)
|
345 |
attn_output = attn_output.reshape(seq_length, -1)
|
346 |
-
attn_output = self.
|
347 |
return attn_output
|
348 |
|
349 |
|
|
|
343 |
attn_output = F.scaled_dot_product_attention(query_states, key_states, value_states, attention_mask, dropout_p=0.0)
|
344 |
attn_output = attn_output.transpose(0, 1)
|
345 |
attn_output = attn_output.reshape(seq_length, -1)
|
346 |
+
attn_output = self.out_proj(attn_output)
|
347 |
return attn_output
|
348 |
|
349 |
|