riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 11:11:05 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

Fix Tensor + Embedding error in some cases when using SiglipVisionModel (#33994)

Browse Source 
Fix Tensor + Embedding error in some cases

Co-authored-by: kaitolucifer <kaito.o@ghelia.com>
This commit is contained in:
Kaito 2024-10-07 18:17:34 +09:00 committed by GitHub
parent 9b4b0c07db
commit e782e95e34
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/models/siglip/modeling_siglip.py
View File

@ -283,7 +283,7 @@ class SiglipVisionEmbeddings(nn.Module):
# always interpolate when tracing to ensure the exported model works for dynamic input shapes # always interpolate when tracing to ensure the exported model works for dynamic input shapes
if not torch.jit.is_tracing() and num_patches == num_positions and height == width: if not torch.jit.is_tracing() and num_patches == num_positions and height == width:
return self.position_embedding return self.position_embedding(self.position_ids)
patch_pos_embed = self.position_embedding patch_pos_embed = self.position_embedding