riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 23 Packages Projects Releases Wiki Activity

Removed unnecessary self.projection call in VivitTubeletEmbeddings (#31632)

Browse Source 
removes unnecessary second projection call
This commit is contained in:
Vladimir Iashin 2024-06-26 11:19:26 +01:00 committed by GitHub
parent 2daf2c3eaa
commit 1218e439b5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/transformers/models/vivit/modeling_vivit.py
View File

@ -76,7 +76,8 @@ class VivitTubeletEmbeddings(nn.Module):
x = self.projection(pixel_values)
# out_batch_size, out_num_channels, out_num_frames, out_height, out_width = x.shape
x = self.projection(pixel_values).flatten(2).transpose(1, 2)
# flattens time and space dimensions, transposes to (out_batch_size, flat_tokens, out_num_channels)
x = x.flatten(2).transpose(1, 2)
return x