DeepSeek/DeepSeek-VL
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepSeek-VL synced 2025-06-26 18:27:43 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: automatically patch collections for >python 3.10 support (#21)

Browse Source 
Co-authored-by: Bo Liu <benjaminliu.eecs@gmail.com>
This commit is contained in:
Nicola Dall'Asen
2024-03-13 15:34:10 +01:00
committed by GitHub
parent 601426030d
commit 717ed7639c
5 changed files with 28 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
deepseek_vl/models/siglip_vit.py
View File

@@ -338,9 +338,9 @@ class VisionTransformer(nn.Module):
self.num_classes = num_classes
self.global_pool = global_pool
self.num_features = (
self.embed_dim
) = embed_dim # num_features for consistency with other models
self.num_features = self.embed_dim = (
embed_dim # num_features for consistency with other models
)
self.num_prefix_tokens = 1 if class_token else 0
self.num_prefix_tokens += reg_tokens
self.num_reg_tokens = reg_tokens