lkhl commited on
Commit
5d426e9
·
verified ·
1 Parent(s): 2f04c2f

Update modeling_videollama3_encoder.py

Browse files
Files changed (1) hide show
  1. modeling_videollama3_encoder.py +1 -1
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.proj(attn_output)
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