Text Generation
Transformers
Safetensors
imp
custom_code
Oyoy1235 commited on
Commit
3c9c49b
1 Parent(s): bf3351d

update encoder_config

Browse files
Files changed (2) hide show
  1. tokenizer_config.json +16 -0
  2. vision_encoder.py +1 -0
tokenizer_config.json CHANGED
@@ -2,6 +2,22 @@
2
  "add_bos_token": false,
3
  "add_prefix_space": false,
4
  "added_tokens_decoder": {
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
5
  "50256": {
6
  "content": "<|endoftext|>",
7
  "lstrip": false,
 
2
  "add_bos_token": false,
3
  "add_prefix_space": false,
4
  "added_tokens_decoder": {
5
+ "50296": {
6
+ "content": "<image>",
7
+ "lstrip": false,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false,
11
+ "special": true
12
+ },
13
+ "50295": {
14
+ "content": "</s>",
15
+ "lstrip": false,
16
+ "normalized": false,
17
+ "rstrip": false,
18
+ "single_word": false,
19
+ "special": true
20
+ },
21
  "50256": {
22
  "content": "<|endoftext|>",
23
  "lstrip": false,
vision_encoder.py CHANGED
@@ -466,6 +466,7 @@ class SiglipMultiheadAttentionPoolingHead(nn.Module):
466
  class SiglipVisionModel(SiglipPreTrainedModel):
467
  config_class = SiglipVisionConfig
468
  main_input_name = "pixel_values"
 
469
 
470
  def __init__(self, config: SiglipVisionConfig):
471
  super().__init__(config)
 
466
  class SiglipVisionModel(SiglipPreTrainedModel):
467
  config_class = SiglipVisionConfig
468
  main_input_name = "pixel_values"
469
+ _no_split_modules = ["SiglipEncoderLayer"]
470
 
471
  def __init__(self, config: SiglipVisionConfig):
472
  super().__init__(config)