Upload folder using huggingface_hub

config.json

@@ -0,0 +1,247 @@
+{
+  "_name_or_path": "checkpoints/Flrq-2-large-combined/",
+  "architectures": [
+    "Florence2ForConditionalGeneration"
+  ],
+  "attention_dropout": 0.0,
+  "auto_map": {
+    "AutoConfig": "configuration_florence2.Florence2Config",
+    "AutoModelForCausalLM": "modeling_florence2.Florence2ForConditionalGeneration"
+  },
+  "bos_token_id": 151643,
+  "eos_token_id": 151645,
+  "hidden_act": "silu",
+  "hidden_size": 1536,
+  "ignore_index": -100,
+  "initializer_range": 0.02,
+  "intermediate_size": 8960,
+  "max_position_embeddings": 32768,
+  "max_window_layers": 21,
+  "model_type": "florence2",
+  "num_attention_heads": 12,
+  "num_hidden_layers": 28,
+  "num_key_value_heads": 2,
+  "pad_token_id": 1,
+  "projection_dim": 1024,
+  "rms_norm_eps": 1e-06,
+  "rope_scaling": null,
+  "rope_theta": 1000000.0,
+  "sliding_window": null,
+  "text_config": {
+    "_attn_implementation_autoset": true,
+    "_name_or_path": "",
+    "add_cross_attention": false,
+    "architectures": [
+      "Qwen2ForCausalLM"
+    ],
+    "attention_dropout": 0.0,
+    "bad_words_ids": null,
+    "begin_suppress_tokens": null,
+    "bos_token_id": 151643,
+    "chunk_size_feed_forward": 0,
+    "cross_attention_hidden_size": null,
+    "decoder_start_token_id": null,
+    "diversity_penalty": 0.0,
+    "do_sample": false,
+    "early_stopping": false,
+    "encoder_no_repeat_ngram_size": 0,
+    "eos_token_id": 151645,
+    "exponential_decay_length_penalty": null,
+    "finetuning_task": null,
+    "forced_bos_token_id": null,
+    "forced_eos_token_id": null,
+    "hidden_act": "silu",
+    "hidden_size": 1536,
+    "id2label": {
+      "0": "LABEL_0",
+      "1": "LABEL_1"
+    },
+    "initializer_range": 0.02,
+    "intermediate_size": 8960,
+    "is_decoder": false,
+    "is_encoder_decoder": false,
+    "label2id": {
+      "LABEL_0": 0,
+      "LABEL_1": 1
+    },
+    "length_penalty": 1.0,
+    "max_length": 20,
+    "max_position_embeddings": 32768,
+    "max_window_layers": 21,
+    "min_length": 0,
+    "model_type": "qwen2",
+    "no_repeat_ngram_size": 0,
+    "num_attention_heads": 12,
+    "num_beam_groups": 1,
+    "num_beams": 1,
+    "num_hidden_layers": 28,
+    "num_key_value_heads": 2,
+    "num_return_sequences": 1,
+    "output_attentions": false,
+    "output_hidden_states": false,
+    "output_scores": false,
+    "pad_token_id": null,
+    "prefix": null,
+    "problem_type": null,
+    "pruned_heads": {},
+    "remove_invalid_values": false,
+    "repetition_penalty": 1.0,
+    "return_dict": true,
+    "return_dict_in_generate": false,
+    "rms_norm_eps": 1e-06,
+    "rope_scaling": null,
+    "rope_theta": 1000000.0,
+    "sep_token_id": null,
+    "sliding_window": null,
+    "suppress_tokens": null,
+    "task_specific_params": null,
+    "temperature": 1.0,
+    "tf_legacy_loss": false,
+    "tie_encoder_decoder": false,
+    "tie_word_embeddings": true,
+    "tokenizer_class": null,
+    "top_k": 50,
+    "top_p": 1.0,
+    "torch_dtype": "bfloat16",
+    "torchscript": false,
+    "typical_p": 1.0,
+    "use_bfloat16": false,
+    "use_cache": true,
+    "use_sliding_window": false,
+    "vocab_size": 151936
+  },
+  "tie_word_embeddings": true,
+  "torch_dtype": "bfloat16",
+  "transformers_version": "4.46.1",
+  "use_cache": false,
+  "use_sliding_window": false,
+  "vision_config": {
+    "_attn_implementation_autoset": false,
+    "_name_or_path": "",
+    "add_cross_attention": false,
+    "architectures": null,
+    "bad_words_ids": null,
+    "begin_suppress_tokens": null,
+    "bos_token_id": null,
+    "chunk_size_feed_forward": 0,
+    "cross_attention_hidden_size": null,
+    "decoder_start_token_id": null,
+    "depths": [
+      1,
+      1,
+      9,
+      1
+    ],
+    "dim_embed": [
+      256,
+      512,
+      1024,
+      2048
+    ],
+    "diversity_penalty": 0.0,
+    "do_sample": false,
+    "drop_path_rate": 0.1,
+    "early_stopping": false,
+    "enable_checkpoint": false,
+    "encoder_no_repeat_ngram_size": 0,
+    "eos_token_id": null,
+    "exponential_decay_length_penalty": null,
+    "finetuning_task": null,
+    "forced_bos_token_id": null,
+    "forced_eos_token_id": null,
+    "id2label": {
+      "0": "LABEL_0",
+      "1": "LABEL_1"
+    },
+    "image_feature_source": [
+      "spatial_avg_pool",
+      "temporal_avg_pool"
+    ],
+    "image_pos_embed": {
+      "max_pos_embeddings": 50,
+      "type": "learned_abs_2d"
+    },
+    "is_decoder": false,
+    "is_encoder_decoder": false,
+    "label2id": {
+      "LABEL_0": 0,
+      "LABEL_1": 1
+    },
+    "length_penalty": 1.0,
+    "max_length": 20,
+    "min_length": 0,
+    "model_type": "davit",
+    "no_repeat_ngram_size": 0,
+    "num_beam_groups": 1,
+    "num_beams": 1,
+    "num_groups": [
+      8,
+      16,
+      32,
+      64
+    ],
+    "num_heads": [
+      8,
+      16,
+      32,
+      64
+    ],
+    "num_return_sequences": 1,
+    "output_attentions": false,
+    "output_hidden_states": false,
+    "output_scores": false,
+    "pad_token_id": null,
+    "patch_padding": [
+      3,
+      1,
+      1,
+      1
+    ],
+    "patch_prenorm": [
+      false,
+      true,
+      true,
+      true
+    ],
+    "patch_size": [
+      7,
+      3,
+      3,
+      3
+    ],
+    "patch_stride": [
+      4,
+      2,
+      2,
+      2
+    ],
+    "prefix": null,
+    "problem_type": null,
+    "projection_dim": 1536,
+    "pruned_heads": {},
+    "remove_invalid_values": false,
+    "repetition_penalty": 1.0,
+    "return_dict": true,
+    "return_dict_in_generate": false,
+    "sep_token_id": null,
+    "suppress_tokens": null,
+    "task_specific_params": null,
+    "temperature": 1.0,
+    "tf_legacy_loss": false,
+    "tie_encoder_decoder": false,
+    "tie_word_embeddings": true,
+    "tokenizer_class": null,
+    "top_k": 50,
+    "top_p": 1.0,
+    "torch_dtype": null,
+    "torchscript": false,
+    "typical_p": 1.0,
+    "use_bfloat16": false,
+    "visual_temporal_embedding": {
+      "max_temporal_embeddings": 100,
+      "type": "COSINE"
+    },
+    "window_size": 12
+  },
+  "vocab_size": 151936
+}

configuration_florence2.py

@@ -0,0 +1,349 @@
+# coding=utf-8
+# Copyright 2024 Microsoft and the HuggingFace Inc. team. All rights reserved.
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+import warnings
+
+""" Florence-2 configuration"""
+
+from typing import Optional
+
+from transformers import AutoConfig
+from transformers.configuration_utils import PretrainedConfig
+from transformers.utils import logging
+from transformers import Qwen2Config
+
+logger = logging.get_logger(__name__)
+
+
+class Florence2VisionConfig(PretrainedConfig):
+    r"""
+    This is the configuration class to store the configuration of a [`Florence2VisionModel`]. It is used to instantiate a Florence2VisionModel
+    according to the specified arguments, defining the model architecture. Instantiating a configuration with the
+    defaults will yield a similar configuration to that of the Florence2VisionModel architecture.
+
+    Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the
+    documentation from [`PretrainedConfig`] for more information.
+
+    Args:
+        drop_path_rate (`float`, *optional*, defaults to 0.1):
+            The dropout rate of the drop path layer.
+        patch_size (`List[int]`, *optional*, defaults to [7, 3, 3, 3]):
+            The patch size of the image.
+        patch_stride (`List[int]`, *optional*, defaults to [4, 2, 2, 2]):
+            The patch stride of the image.
+        patch_padding (`List[int]`, *optional*, defaults to [3, 1, 1, 1]):
+            The patch padding of the image.
+        patch_prenorm (`List[bool]`, *optional*, defaults to [false, true, true, true]):
+            Whether to apply layer normalization before the patch embedding layer.
+        enable_checkpoint (`bool`, *optional*, defaults to False):
+            Whether to enable checkpointing.
+        dim_embed (`List[int]`, *optional*, defaults to [256, 512, 1024, 2048]):
+            The dimension of the embedding layer.
+        num_heads (`List[int]`, *optional*, defaults to [8, 16, 32, 64]):
+            The number of attention heads.
+        num_groups (`List[int]`, *optional*, defaults to [8, 16, 32, 64]):
+            The number of groups.
+        depths (`List[int]`, *optional*, defaults to [1, 1, 9, 1]):
+            The depth of the model.
+        window_size (`int`, *optional*, defaults to 12):
+            The window size of the model.
+        projection_dim (`int`, *optional*, defaults to 1024):
+            The dimension of the projection layer.
+        visual_temporal_embedding (`dict`, *optional*):
+            The configuration of the visual temporal embedding.
+        image_pos_embed (`dict`, *optional*):
+            The configuration of the image position embedding.
+        image_feature_source (`List[str]`, *optional*, defaults to ["spatial_avg_pool", "temporal_avg_pool"]):
+            The source of the image feature.
+    Example:
+
+    ```python
+    >>> from transformers import Florence2VisionConfig, Florence2VisionModel
+
+    >>> # Initializing a Florence2 Vision style configuration
+    >>> configuration = Florence2VisionConfig()
+
+    >>> # Initializing a model (with random weights)
+    >>> model = Florence2VisionModel(configuration)
+
+    >>> # Accessing the model configuration
+    >>> configuration = model.config
+    ```"""
+
+    model_type = "florence2_vision"
+    keys_to_ignore_at_inference = ["past_key_values"]
+
+    def __init__(
+        self,
+        drop_path_rate=0.1,
+        patch_size=[7, 3, 3, 3],
+        patch_stride=[4, 2, 2, 2],
+        patch_padding=[3, 1, 1, 1],
+        patch_prenorm=[False, True, True, True],
+        enable_checkpoint=False,
+        dim_embed=[256, 512, 1024, 2048],
+        num_heads=[8, 16, 32, 64],
+        num_groups=[8, 16, 32, 64],
+        depths=[1, 1, 9, 1],
+        window_size=12,
+        projection_dim=1024,
+        visual_temporal_embedding=None,
+        image_pos_embed=None,
+        image_feature_source=["spatial_avg_pool", "temporal_avg_pool"],
+        **kwargs,
+    ):
+        self.drop_path_rate = drop_path_rate
+        self.patch_size = patch_size
+        self.patch_stride = patch_stride
+        self.patch_padding = patch_padding
+        self.patch_prenorm = patch_prenorm
+        self.enable_checkpoint = enable_checkpoint
+        self.dim_embed = dim_embed
+        self.num_heads = num_heads
+        self.num_groups = num_groups
+        self.depths = depths
+        self.window_size = window_size
+        self.projection_dim = projection_dim
+        self.visual_temporal_embedding = visual_temporal_embedding
+        self.image_pos_embed = image_pos_embed
+        self.image_feature_source = image_feature_source
+
+        super().__init__(**kwargs)
+
+
+class Florence2LanguageConfig(PretrainedConfig):
+    r"""
+    This is the configuration class to store the configuration of a [`Florence2LanguagePreTrainedModel`]. It is used to instantiate a BART
+    model according to the specified arguments, defining the model architecture. Instantiating a configuration with the
+    defaults will yield a similar configuration to that of the BART
+    [facebook/bart-large](https://huggingface.co/facebook/bart-large) architecture.
+
+    Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the
+    documentation from [`PretrainedConfig`] for more information.
+
+
+    Args:
+        vocab_size (`int`, *optional*, defaults to 51289):
+            Vocabulary size of the Florence2Language model. Defines the number of different tokens that can be represented by the
+            `inputs_ids` passed when calling [`Florence2LanguageModel`].
+        d_model (`int`, *optional*, defaults to 1024):
+            Dimensionality of the layers and the pooler layer.
+        encoder_layers (`int`, *optional*, defaults to 12):
+            Number of encoder layers.
+        decoder_layers (`int`, *optional*, defaults to 12):
+            Number of decoder layers.
+        encoder_attention_heads (`int`, *optional*, defaults to 16):
+            Number of attention heads for each attention layer in the Transformer encoder.
+        decoder_attention_heads (`int`, *optional*, defaults to 16):
+            Number of attention heads for each attention layer in the Transformer decoder.
+        decoder_ffn_dim (`int`, *optional*, defaults to 4096):
+            Dimensionality of the "intermediate" (often named feed-forward) layer in decoder.
+        encoder_ffn_dim (`int`, *optional*, defaults to 4096):
+            Dimensionality of the "intermediate" (often named feed-forward) layer in decoder.
+        activation_function (`str` or `function`, *optional*, defaults to `"gelu"`):
+            The non-linear activation function (function or string) in the encoder and pooler. If string, `"gelu"`,
+            `"relu"`, `"silu"` and `"gelu_new"` are supported.
+        dropout (`float`, *optional*, defaults to 0.1):
+            The dropout probability for all fully connected layers in the embeddings, encoder, and pooler.
+        attention_dropout (`float`, *optional*, defaults to 0.0):
+            The dropout ratio for the attention probabilities.
+        activation_dropout (`float`, *optional*, defaults to 0.0):
+            The dropout ratio for activations inside the fully connected layer.
+        classifier_dropout (`float`, *optional*, defaults to 0.0):
+            The dropout ratio for classifier.
+        max_position_embeddings (`int`, *optional*, defaults to 1024):
+            The maximum sequence length that this model might ever be used with. Typically set this to something large
+            just in case (e.g., 512 or 1024 or 2048).
+        init_std (`float`, *optional*, defaults to 0.02):
+            The standard deviation of the truncated_normal_initializer for initializing all weight matrices.
+        encoder_layerdrop (`float`, *optional*, defaults to 0.0):
+            The LayerDrop probability for the encoder. See the [LayerDrop paper](see https://arxiv.org/abs/1909.11556)
+            for more details.
+        decoder_layerdrop (`float`, *optional*, defaults to 0.0):
+            The LayerDrop probability for the decoder. See the [LayerDrop paper](see https://arxiv.org/abs/1909.11556)
+            for more details.
+        scale_embedding (`bool`, *optional*, defaults to `False`):
+            Scale embeddings by diving by sqrt(d_model).
+        use_cache (`bool`, *optional*, defaults to `True`):
+            Whether or not the model should return the last key/values attentions (not used by all models).
+        num_labels (`int`, *optional*, defaults to 3):
+            The number of labels to use in [`Florence2LanguageForSequenceClassification`].
+        forced_eos_token_id (`int`, *optional*, defaults to 2):
+            The id of the token to force as the last generated token when `max_length` is reached. Usually set to
+            `eos_token_id`.
+
+    Example:
+
+    ```python
+    >>> from transformers import Florence2LanguageConfig, Florence2LanguageModel
+
+    >>> # Initializing a Florence2 Language style configuration
+    >>> configuration = Florence2LanguageConfig()
+
+    >>> # Initializing a model (with random weights)
+    >>> model = Florence2LangaugeModel(configuration)
+
+    >>> # Accessing the model configuration
+    >>> configuration = model.config
+    ```"""
+
+    model_type = "florence2_language"
+    keys_to_ignore_at_inference = ["past_key_values"]
+    attribute_map = {
+        "num_attention_heads": "encoder_attention_heads",
+        "hidden_size": "d_model",
+    }
+
+    def __init__(
+        self,
+        vocab_size=51289,
+        max_position_embeddings=1024,
+        encoder_layers=12,
+        encoder_ffn_dim=4096,
+        encoder_attention_heads=16,
+        decoder_layers=12,
+        decoder_ffn_dim=4096,
+        decoder_attention_heads=16,
+        encoder_layerdrop=0.0,
+        decoder_layerdrop=0.0,
+        activation_function="gelu",
+        d_model=1024,
+        dropout=0.1,
+        attention_dropout=0.0,
+        activation_dropout=0.0,
+        init_std=0.02,
+        classifier_dropout=0.0,
+        scale_embedding=False,
+        use_cache=True,
+        num_labels=3,
+        pad_token_id=1,
+        bos_token_id=0,
+        eos_token_id=2,
+        is_encoder_decoder=True,
+        decoder_start_token_id=2,
+        forced_eos_token_id=2,
+        **kwargs,
+    ):
+        self.vocab_size = vocab_size
+        self.max_position_embeddings = max_position_embeddings
+        self.d_model = d_model
+        self.encoder_ffn_dim = encoder_ffn_dim
+        self.encoder_layers = encoder_layers
+        self.encoder_attention_heads = encoder_attention_heads
+        self.decoder_ffn_dim = decoder_ffn_dim
+        self.decoder_layers = decoder_layers
+        self.decoder_attention_heads = decoder_attention_heads
+        self.dropout = dropout
+        self.attention_dropout = attention_dropout
+        self.activation_dropout = activation_dropout
+        self.activation_function = activation_function
+        self.init_std = init_std
+        self.encoder_layerdrop = encoder_layerdrop
+        self.decoder_layerdrop = decoder_layerdrop
+        self.classifier_dropout = classifier_dropout
+        self.use_cache = use_cache
+        self.num_hidden_layers = encoder_layers
+        self.scale_embedding = (
+            scale_embedding  # scale factor will be sqrt(d_model) if True
+        )
+
+        super().__init__(
+            num_labels=num_labels,
+            pad_token_id=pad_token_id,
+            bos_token_id=bos_token_id,
+            eos_token_id=eos_token_id,
+            is_encoder_decoder=is_encoder_decoder,
+            decoder_start_token_id=decoder_start_token_id,
+            forced_eos_token_id=forced_eos_token_id,
+            **kwargs,
+        )
+
+        # ensure backward compatibility for BART CNN models
+        if self.forced_bos_token_id is None and kwargs.get(
+            "force_bos_token_to_be_generated", False
+        ):
+            self.forced_bos_token_id = self.bos_token_id
+            warnings.warn(
+                f"Please make sure the config includes `forced_bos_token_id={self.bos_token_id}` in future versions. "
+                "The config can simply be saved and uploaded again to be fixed."
+            )
+
+
+class Florence2Config(Qwen2Config):
+    r"""
+    This is the configuration class to store the configuration of a [`Florence2ForConditionalGeneration`]. It is used to instantiate an
+    Florence-2 model according to the specified arguments, defining the model architecture.
+
+    Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the
+    documentation from [`PretrainedConfig`] for more information.
+
+    Args:
+        vision_config (`Florence2VisionConfig`,  *optional*):
+            Custom vision config or dict
+        text_config (`Union[AutoConfig, dict]`, *optional*):
+            The config object of the text backbone.
+        ignore_index (`int`, *optional*, defaults to -100):
+            The ignore index for the loss function.
+        vocab_size (`int`, *optional*, defaults to 51289):
+            Vocabulary size of the Florence2model. Defines the number of different tokens that can be represented by the
+            `inputs_ids` passed when calling [`~Florence2ForConditionalGeneration`]
+        projection_dim (`int`, *optional*, defaults to 1024):
+            Dimension of the multimodal projection space.
+
+    Example:
+
+    ```python
+    >>> from transformers import Florence2ForConditionalGeneration, Florence2Config, CLIPVisionConfig, BartConfig
+
+    >>> # Initializing a clip-like vision config
+    >>> vision_config = CLIPVisionConfig()
+
+    >>> # Initializing a Bart config
+    >>> text_config = BartConfig()
+
+    >>> # Initializing a Florence-2 configuration
+    >>> configuration = Florence2Config(vision_config, text_config)
+
+    >>> # Initializing a model from the florence-2 configuration
+    >>> model = Florence2ForConditionalGeneration(configuration)
+
+    >>> # Accessing the model configuration
+    >>> configuration = model.config
+    ```"""
+
+    model_type = "florence2"
+    is_composition = False
+
+    def __init__(
+        self,
+        vision_config=None,
+        text_config=None,
+        ignore_index=-100,
+        vocab_size=51289,
+        projection_dim=1024,
+        **kwargs,
+    ):
+        self.ignore_index = ignore_index
+        self.vocab_size = vocab_size
+        self.projection_dim = projection_dim
+        if vision_config is not None:
+            vision_config = PretrainedConfig(**vision_config)
+        self.vision_config = vision_config
+        self.vocab_size = self.vocab_size
+
+        self.text_config = text_config
+        if text_config is not None:
+            # self.text_config = Florence2LanguageConfig(**text_config)
+            self.text_config = Qwen2Config(**text_config)
+
+        super().__init__(**kwargs)

generation_config.json

@@ -0,0 +1,7 @@
+{
+  "_from_model_config": true,
+  "bos_token_id": 151643,
+  "eos_token_id": 151645,
+  "pad_token_id": 1,
+  "transformers_version": "4.46.1"
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1646640fb73d58ca5877fb55d5eb9cce8b69d0e3889d42646dcb3c1b865e1da2
+size 3815697496

model.safetensors.index.json

@@ -0,0 +1,751 @@
+{
+  "metadata": {
+    "total_size": 7631210496
+  },
+  "weight_map": {
+    "image_pos_embed.column_embeddings.weight": "model-00002-of-00002.safetensors",
+    "image_pos_embed.row_embeddings.weight": "model-00002-of-00002.safetensors",
+    "image_proj_norm.bias": "model-00002-of-00002.safetensors",
+    "image_proj_norm.weight": "model-00002-of-00002.safetensors",
+    "image_projection": "model-00001-of-00002.safetensors",
+    "model.embed_tokens.weight": "model-00001-of-00002.safetensors",
+    "model.layers.0.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.0.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.0.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.0.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.0.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.0.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.1.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.1.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.1.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.1.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.1.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.1.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.10.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.10.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.10.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.10.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.10.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.10.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.10.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.10.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.10.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.11.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.11.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.11.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.11.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.11.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.11.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.11.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.11.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.11.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.11.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.11.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.11.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.12.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.12.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.12.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.12.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.12.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.12.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.12.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.12.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.12.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.12.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.12.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.12.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.13.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.13.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.13.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.13.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.13.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.13.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.13.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.13.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.13.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.13.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.13.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.13.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.14.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.14.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.14.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.14.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.14.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.14.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.14.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.14.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.14.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.14.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.14.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.14.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.15.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.15.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.15.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.15.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.15.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.15.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.15.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.15.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.15.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.15.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.15.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.15.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.16.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.16.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.16.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.16.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.16.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.16.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.16.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.16.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.16.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.16.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.16.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.16.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.17.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.17.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.17.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.17.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.17.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.17.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.17.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.17.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.17.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.17.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.17.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.17.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.18.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.18.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.18.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.18.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.18.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.18.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.18.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.18.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.18.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.18.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.18.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.18.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.19.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.19.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.19.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.19.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.19.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.19.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.19.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.19.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.19.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.19.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.19.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.19.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.2.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.2.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.2.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.2.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.2.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.2.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.20.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.20.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.20.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.20.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.20.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.20.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.20.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.20.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.20.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.20.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.20.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.20.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.21.input_layernorm.weight": "model-00002-of-00002.safetensors",
+    "model.layers.21.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.21.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.21.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.21.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
+    "model.layers.21.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.21.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.21.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.21.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.21.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.21.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.21.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.22.input_layernorm.weight": "model-00002-of-00002.safetensors",
+    "model.layers.22.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.22.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.22.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.22.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
+    "model.layers.22.self_attn.k_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.22.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.22.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.22.self_attn.q_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.22.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.22.self_attn.v_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.22.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.23.input_layernorm.weight": "model-00002-of-00002.safetensors",
+    "model.layers.23.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.23.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.23.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.23.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
+    "model.layers.23.self_attn.k_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.23.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.23.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.23.self_attn.q_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.23.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.23.self_attn.v_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.23.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.24.input_layernorm.weight": "model-00002-of-00002.safetensors",
+    "model.layers.24.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.24.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.24.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.24.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
+    "model.layers.24.self_attn.k_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.24.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.24.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.24.self_attn.q_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.24.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.24.self_attn.v_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.24.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.25.input_layernorm.weight": "model-00002-of-00002.safetensors",
+    "model.layers.25.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.25.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.25.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.25.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
+    "model.layers.25.self_attn.k_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.25.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.25.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.25.self_attn.q_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.25.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.25.self_attn.v_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.25.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.26.input_layernorm.weight": "model-00002-of-00002.safetensors",
+    "model.layers.26.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.26.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.26.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.26.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
+    "model.layers.26.self_attn.k_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.26.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.26.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.26.self_attn.q_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.26.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.26.self_attn.v_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.26.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.27.input_layernorm.weight": "model-00002-of-00002.safetensors",
+    "model.layers.27.mlp.down_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.27.mlp.gate_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.27.mlp.up_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.27.post_attention_layernorm.weight": "model-00002-of-00002.safetensors",
+    "model.layers.27.self_attn.k_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.27.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.27.self_attn.o_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.27.self_attn.q_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.27.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.27.self_attn.v_proj.bias": "model-00002-of-00002.safetensors",
+    "model.layers.27.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
+    "model.layers.3.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.3.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.3.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.3.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.3.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.3.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.4.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.4.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.4.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.4.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.4.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.4.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.5.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.5.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.5.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.5.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.5.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.5.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.6.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.6.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.6.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.6.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.6.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.6.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.7.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.7.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.7.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.7.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.7.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.7.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.8.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.8.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.8.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.8.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.8.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.8.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.8.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.8.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.9.input_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.9.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.9.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.9.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.9.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+    "model.layers.9.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.9.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.9.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+    "model.layers.9.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
+    "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+    "model.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.channel_block.channel_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.channel_block.channel_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.channel_block.channel_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.channel_block.channel_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.channel_block.channel_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.channel_block.channel_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.channel_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.channel_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.channel_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.channel_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.channel_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.channel_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.channel_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.channel_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.channel_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.channel_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.spatial_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.spatial_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.spatial_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.spatial_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.spatial_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.spatial_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.spatial_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.spatial_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.spatial_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.spatial_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.spatial_block.window_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.spatial_block.window_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.spatial_block.window_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.spatial_block.window_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.spatial_block.window_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.0.0.spatial_block.window_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.channel_block.channel_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.channel_block.channel_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.channel_block.channel_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.channel_block.channel_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.channel_block.channel_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.channel_block.channel_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.channel_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.channel_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.channel_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.channel_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.channel_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.channel_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.channel_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.channel_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.channel_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.channel_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.spatial_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.spatial_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.spatial_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.spatial_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.spatial_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.spatial_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.spatial_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.spatial_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.spatial_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.spatial_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.spatial_block.window_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.spatial_block.window_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.spatial_block.window_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.spatial_block.window_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.spatial_block.window_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.1.0.spatial_block.window_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.channel_block.channel_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.channel_block.channel_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.channel_block.channel_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.channel_block.channel_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.channel_block.channel_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.channel_block.channel_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.channel_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.channel_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.channel_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.channel_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.channel_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.channel_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.channel_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.channel_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.channel_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.channel_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.spatial_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.spatial_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.spatial_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.spatial_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.spatial_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.spatial_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.spatial_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.spatial_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.spatial_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.spatial_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.spatial_block.window_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.spatial_block.window_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.spatial_block.window_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.spatial_block.window_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.spatial_block.window_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.0.spatial_block.window_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.channel_block.channel_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.channel_block.channel_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.channel_block.channel_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.channel_block.channel_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.channel_block.channel_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.channel_block.channel_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.channel_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.channel_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.channel_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.channel_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.channel_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.channel_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.channel_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.channel_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.channel_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.channel_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.spatial_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.spatial_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.spatial_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.spatial_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.spatial_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.spatial_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.spatial_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.spatial_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.spatial_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.spatial_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.spatial_block.window_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.spatial_block.window_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.spatial_block.window_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.spatial_block.window_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.spatial_block.window_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.1.spatial_block.window_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.channel_block.channel_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.channel_block.channel_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.channel_block.channel_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.channel_block.channel_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.channel_block.channel_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.channel_block.channel_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.channel_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.channel_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.channel_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.channel_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.channel_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.channel_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.channel_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.channel_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.channel_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.channel_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.spatial_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.spatial_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.spatial_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.spatial_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.spatial_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.spatial_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.spatial_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.spatial_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.spatial_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.spatial_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.spatial_block.window_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.spatial_block.window_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.spatial_block.window_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.spatial_block.window_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.spatial_block.window_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.2.spatial_block.window_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.channel_block.channel_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.channel_block.channel_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.channel_block.channel_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.channel_block.channel_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.channel_block.channel_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.channel_block.channel_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.channel_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.channel_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.channel_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.channel_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.channel_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.channel_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.channel_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.channel_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.channel_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.channel_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.spatial_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.spatial_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.spatial_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.spatial_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.spatial_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.spatial_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.spatial_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.spatial_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.spatial_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.spatial_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.spatial_block.window_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.spatial_block.window_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.spatial_block.window_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.spatial_block.window_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.spatial_block.window_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.3.spatial_block.window_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.channel_block.channel_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.channel_block.channel_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.channel_block.channel_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.channel_block.channel_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.channel_block.channel_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.channel_block.channel_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.channel_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.channel_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.channel_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.channel_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.channel_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.channel_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.channel_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.channel_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.channel_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.channel_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.spatial_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.spatial_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.spatial_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.spatial_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.spatial_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.spatial_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.spatial_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.spatial_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.spatial_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.spatial_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.spatial_block.window_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.spatial_block.window_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.spatial_block.window_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.spatial_block.window_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.spatial_block.window_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.4.spatial_block.window_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.channel_block.channel_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.channel_block.channel_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.channel_block.channel_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.channel_block.channel_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.channel_block.channel_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.channel_block.channel_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.channel_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.channel_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.channel_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.channel_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.channel_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.channel_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.channel_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.channel_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.channel_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.channel_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.spatial_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.spatial_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.spatial_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.spatial_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.spatial_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.spatial_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.spatial_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.spatial_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.spatial_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.spatial_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.spatial_block.window_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.spatial_block.window_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.spatial_block.window_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.spatial_block.window_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.spatial_block.window_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.5.spatial_block.window_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.channel_block.channel_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.channel_block.channel_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.channel_block.channel_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.channel_block.channel_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.channel_block.channel_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.channel_block.channel_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.channel_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.channel_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.channel_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.channel_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.channel_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.channel_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.channel_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.channel_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.channel_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.channel_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.spatial_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.spatial_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.spatial_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.spatial_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.spatial_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.spatial_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.spatial_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.spatial_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.spatial_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.spatial_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.spatial_block.window_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.spatial_block.window_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.spatial_block.window_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.spatial_block.window_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.spatial_block.window_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.6.spatial_block.window_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.channel_block.channel_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.channel_block.channel_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.channel_block.channel_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.channel_block.channel_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.channel_block.channel_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.channel_block.channel_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.channel_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.channel_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.channel_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.channel_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.channel_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.channel_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.channel_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.channel_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.channel_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.channel_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.spatial_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.spatial_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.spatial_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.spatial_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.spatial_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.spatial_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.spatial_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.spatial_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.spatial_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.spatial_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.spatial_block.window_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.spatial_block.window_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.spatial_block.window_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.spatial_block.window_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.spatial_block.window_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.7.spatial_block.window_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.channel_block.channel_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.channel_block.channel_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.channel_block.channel_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.channel_block.channel_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.channel_block.channel_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.channel_block.channel_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.channel_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.channel_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.channel_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.channel_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.channel_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.channel_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.channel_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.channel_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.channel_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.channel_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.spatial_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.spatial_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.spatial_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.spatial_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.spatial_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.spatial_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.spatial_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.spatial_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.spatial_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.spatial_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.spatial_block.window_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.spatial_block.window_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.spatial_block.window_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.spatial_block.window_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.spatial_block.window_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.2.8.spatial_block.window_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.channel_block.channel_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.channel_block.channel_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.channel_block.channel_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.channel_block.channel_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.channel_block.channel_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.channel_block.channel_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.channel_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.channel_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.channel_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.channel_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.channel_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.channel_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.channel_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.channel_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.channel_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.channel_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.spatial_block.conv1.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.spatial_block.conv1.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.spatial_block.conv2.fn.dw.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.spatial_block.conv2.fn.dw.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.spatial_block.ffn.fn.net.fc1.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.spatial_block.ffn.fn.net.fc1.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.spatial_block.ffn.fn.net.fc2.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.spatial_block.ffn.fn.net.fc2.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.spatial_block.ffn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.spatial_block.ffn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.spatial_block.window_attn.fn.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.spatial_block.window_attn.fn.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.spatial_block.window_attn.fn.qkv.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.spatial_block.window_attn.fn.qkv.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.spatial_block.window_attn.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.blocks.3.0.spatial_block.window_attn.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.convs.0.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.convs.0.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.convs.0.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.convs.0.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.convs.1.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.convs.1.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.convs.1.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.convs.1.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.convs.2.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.convs.2.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.convs.2.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.convs.2.proj.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.convs.3.norm.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.convs.3.norm.weight": "model-00002-of-00002.safetensors",
+    "vision_tower.convs.3.proj.bias": "model-00002-of-00002.safetensors",
+    "vision_tower.convs.3.proj.weight": "model-00002-of-00002.safetensors",
+    "visual_temporal_embed.pos_idx_to_embed": "model-00002-of-00002.safetensors"
+  }
+}

preprocessor_config.json

@@ -0,0 +1,39 @@
+{
+  "auto_map": {
+    "AutoProcessor": "processing_florence2.Florence2Processor"
+   },
+  "_valid_processor_keys": [
+    "images",
+    "do_resize",
+    "size",
+    "resample",
+    "do_rescale",
+    "rescale_factor",
+    "do_normalize",
+    "image_mean",
+    "image_std",
+    "return_tensors",
+    "data_format",
+    "input_data_format",
+    "do_convert_rgb"
+  ],
+  "do_convert_rgb": null,
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "do_center_crop": false,
+  "image_processor_type": "CLIPImageProcessor",
+  "image_seq_length": 577,
+  "image_mean": [0.485, 0.456, 0.406],
+  "image_std":  [0.229, 0.224, 0.225],
+  "processor_class": "Florence2Processor",
+  "resample": 3,
+  "size": {
+    "height": 768,
+    "width":768 
+  },
+  "crop_size": {
+    "height": 768,
+    "width": 768
+  }
+}

processing_florence2.py

@@ -0,0 +1,1135 @@
+# coding=utf-8
+# Copyright 2024 Microsoft and The HuggingFace Inc. team.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+"""
+Processor class for Florence-2.
+"""
+
+import re
+import logging
+from typing import List, Optional, Union
+import numpy as np
+
+import torch
+
+from transformers.feature_extraction_utils import BatchFeature
+from transformers.image_utils import ImageInput, is_valid_image
+from transformers.processing_utils import ProcessorMixin
+from transformers.tokenization_utils_base import (
+    PaddingStrategy,
+    PreTokenizedInput,
+    TextInput,
+    TruncationStrategy,
+)
+from transformers.utils import TensorType
+
+IMAGE_TOKEN_INDEX = -200
+
+logger = logging.getLogger(__name__)
+
+# Copied from transformers.models.idefics2.processing_idefics2.is_url
+def is_url(val) -> bool:
+    return isinstance(val, str) and val.startswith("http")
+
+# Copied from transformers.models.idefics2.processing_idefics2.is_image_or_image_url
+def is_image_or_image_url(elem):
+    return is_url(elem) or is_valid_image(elem)
+
+
+def _is_str_or_image(elem):
+    return isinstance(elem, (str)) or is_image_or_image_url(elem)
+
+
+class Florence2Processor(ProcessorMixin):
+    r"""
+    Constructs a Florence2 processor which wraps a Florence2 image processor and a Florence2 tokenizer into a single processor.
+
+    [`Florence2Processor`] offers all the functionalities of [`CLIPImageProcessor`] and [`BartTokenizerFast`]. See the
+    [`~Florence2Processor.__call__`] and [`~Florence2Processor.decode`] for more information.
+
+    Args:
+        image_processor ([`CLIPImageProcessor`], *optional*):
+            The image processor is a required input.
+        tokenizer ([`BartTokenizerFast`], *optional*):
+            The tokenizer is a required input.
+    """
+
+    attributes = ["image_processor", "tokenizer"]
+    image_processor_class = "CLIPImageProcessor"
+    tokenizer_class = ("Qwen2Tokenizer", "Qwen2TokenizerFast")
+
+    def __init__(
+        self,
+        image_processor=None,
+        tokenizer=None,
+    ):
+        if image_processor is None:
+            raise ValueError("You need to specify an `image_processor`.")
+        if tokenizer is None:
+            raise ValueError("You need to specify a `tokenizer`.")
+        if not hasattr(image_processor, "image_seq_length"):
+            raise ValueError("Image processor is missing an `image_seq_length` attribute.")
+
+        self.image_seq_length = image_processor.image_seq_length
+
+        tokens_to_add = {
+                'additional_special_tokens': \
+                    tokenizer.additional_special_tokens + \
+                    ['<od>', '</od>', '<ocr>', '</ocr>'] + \
+                    [f'<loc_{x}>' for x in range(1000)] + \
+                    ['<cap>', '</cap>', '<ncap>', '</ncap>','<dcap>', '</dcap>', '<grounding>', '</grounding>', '<seg>', '</seg>', '<sep>', '<region_cap>', '</region_cap>', '<region_to_desciption>', '</region_to_desciption>', '<proposal>', '</proposal>', '<poly>', '</poly>', '<and>']
+            }
+        tokenizer.add_special_tokens(tokens_to_add)
+
+        self.tasks_answer_post_processing_type = {
+            '<OCR>': 'pure_text',
+            '<OCR_TABLE_MD>': 'pure_text',
+            '<OCR_WITH_REGION>': 'ocr',
+            '<CAPTION>': 'pure_text',
+            '<CAPTION_CN>': 'pure_text',
+            '<DETAILED_CAPTION>': 'pure_text',
+            '<DETAILED_CAPTION_CN>': 'pure_text',
+            '<MORE_DETAILED_CAPTION>': 'pure_text',
+            '<MORE_DETAILED_CAPTION_CN>': 'pure_text',
+            '<OD>': 'description_with_bboxes',
+            '<DENSE_REGION_CAPTION>': 'description_with_bboxes',
+            '<CAPTION_TO_PHRASE_GROUNDING>': "phrase_grounding",
+            '<REFERRING_EXPRESSION_SEGMENTATION>': 'polygons',
+            '<REGION_TO_SEGMENTATION>': 'polygons',
+            '<OPEN_VOCABULARY_DETECTION>': 'description_with_bboxes_or_polygons',
+            '<REGION_TO_CATEGORY>': 'pure_text',
+            '<REGION_TO_DESCRIPTION>': 'pure_text',
+            '<REGION_TO_OCR>': 'pure_text',
+            '<REGION_PROPOSAL>': 'bboxes'
+        }
+
+        self.task_prompts_without_inputs = {
+            '<OCR>': 'What is the text in the image?',
+            '<OCR_WITH_REGION>': 'What is the text in the image, with regions?',
+            '<OCR_TABLE_MD>': 'Convert the image into markdown style reconstructed table in text.',
+            '<CAPTION>': 'What does the image describe?',
+            '<CAPTION_CN>': '图中描述了什么？',
+            '<DETAILED_CAPTION>': 'Describe in detail what is shown in the image.',
+            '<DETAILED_CAPTION_CN>': '详细描述图像中所展示的内容。',
+            '<MORE_DETAILED_CAPTION>': 'Describe with a paragraph what is shown in the image.',
+            '<MORE_DETAILED_CAPTION_CN>': '用一段话描述图像中所展示的内容。',
+            '<OD>': 'Locate the objects with category name in the image.',
+            '<DENSE_REGION_CAPTION>': 'Locate the objects in the image, with their descriptions.',
+            '<REGION_PROPOSAL>': 'Locate the region proposals in the image.'
+        }
+
+        self.task_prompts_with_input = {
+            '<CAPTION_TO_PHRASE_GROUNDING>': "Locate the phrases in the caption: {input}",
+            '<REFERRING_EXPRESSION_SEGMENTATION>': 'Locate {input} in the image with mask',
+            '<REGION_TO_SEGMENTATION>': 'What is the polygon mask of region {input}',
+            '<OPEN_VOCABULARY_DETECTION>': 'Locate {input} in the image.',
+            '<REGION_TO_CATEGORY>': 'What is the region {input}?',
+            '<REGION_TO_DESCRIPTION>': 'What does the region {input} describe?',
+            '<REGION_TO_OCR>': 'What text is in the region {input}?',
+        }
+
+        self.post_processor = Florence2PostProcesser(tokenizer=tokenizer)
+
+
+        super().__init__(image_processor, tokenizer)
+    
+    def _construct_prompts(self, text):
+        # replace the task tokens with the task prompts if task token is in the text
+        sys_prompt = '<|im_start|>system\nYou are a helpful AI assistant<|im_end|>\n'
+        prompts = []
+        for _text in text:
+            if _text in self.task_prompts_without_inputs.keys():
+                # 1. fixed task prompts without additional inputs
+                for task_token, task_prompt in self.task_prompts_without_inputs.items():
+                    if task_token in _text:
+                        assert _text == task_token, f"Task token {task_token} should be the only token in the text."
+                        _text = task_prompt
+                        break
+                # 2. task prompts with additional inputs 
+                for task_token, task_prompt in self.task_prompts_with_input.items():
+                    if task_token in _text:
+                        _text = task_prompt.format(input=_text.replace(task_token, ''))
+                        break
+                # add prompt template here
+                _text = f'{sys_prompt}<|im_start|>user\n<image>\n{_text}<|im_end|>\n<|im_start|>assistant:\n'
+            else:
+                _text = f'{sys_prompt}<|im_start|>user\n{_text}<|im_end|>\n<|im_start|>assistant:\n'
+            prompts.append(_text)
+        return prompts
+
+    def tokenizer_image_token(self, prompt, tokenizer, image_token_index=IMAGE_TOKEN_INDEX, return_tensors=None, add_special_tokens=True):
+        prompt_chunks = [tokenizer(chunk, add_special_tokens=add_special_tokens).input_ids for chunk in prompt.split('<image>')]
+
+        def insert_separator(X, sep):
+            return [ele for sublist in zip(X, [sep]*len(X)) for ele in sublist][:-1]
+
+        input_ids = []
+        offset = 0
+        if len(prompt_chunks) > 0 and len(prompt_chunks[0]) > 0 and prompt_chunks[0][0] == tokenizer.bos_token_id:
+            offset = 1
+            input_ids.append(prompt_chunks[0][0])
+
+        for x in insert_separator(prompt_chunks, [image_token_index] * (offset + 1)):
+            input_ids.extend(x[offset:])
+
+        if return_tensors is not None:
+            if return_tensors == 'pt':
+                return torch.tensor(input_ids, dtype=torch.long)
+            raise ValueError(f'Unsupported tensor type: {return_tensors}')
+        return input_ids
+
+    def __call__(
+        self,
+        text: Union[TextInput, PreTokenizedInput, List[TextInput], List[PreTokenizedInput]] = None,
+        images: ImageInput = None,
+        tokenize_newline_separately: bool = True,
+        padding: Union[bool, str, PaddingStrategy] = False,
+        truncation: Union[bool, str, TruncationStrategy] = None,
+        max_length=None,
+        return_tensors: Optional[Union[str, TensorType]] = TensorType.PYTORCH,
+        do_resize: bool = None,
+        do_normalize: bool = None,
+        image_mean: Optional[Union[float, List[float]]] = None,
+        image_std: Optional[Union[float, List[float]]] = None,
+        data_format: Optional["ChannelDimension"] = "channels_first",  # noqa: F821
+        input_data_format: Optional[
+            Union[str, "ChannelDimension"]  # noqa: F821
+        ] = None,
+        resample: "PILImageResampling" = None,  # noqa: F821
+        size=None,
+        do_convert_rgb: bool = None,
+        do_thumbnail: bool = None,
+        do_align_long_axis: bool = None,
+        do_rescale: bool = None,
+    ) -> BatchFeature:
+        """
+        Main method to prepare for the model one or several sequences(s) and image(s). This method forwards the `text`
+        and `kwargs` arguments to BartTokenizerFast's [`~BartTokenizerFast.__call__`] if `text` is not `None` to encode
+        the text. To prepare the image(s), this method forwards the `images` and `kwrags` arguments to
+        CLIPImageProcessor's [`~CLIPImageProcessor.__call__`] if `images` is not `None`. Please refer to the doctsring
+        of the above two methods for more information.
+
+        Args:
+            text (`str`, `List[str]`, `List[List[str]]`):
+                The sequence or batch of sequences to be encoded. Each sequence can be a string or a list of strings
+                (pretokenized string). If the sequences are provided as list of strings (pretokenized), you must set
+                `is_split_into_words=True` (to lift the ambiguity with a batch of sequences).
+            images (`PIL.Image.Image`, `np.ndarray`, `torch.Tensor`, `List[PIL.Image.Image]`, `List[np.ndarray]`, `List[torch.Tensor]`):
+                The image or batch of images to be prepared. Each image can be a PIL image, NumPy array or PyTorch
+                tensor. In case of a NumPy array/PyTorch tensor, each image should be of shape (C, H, W), where C is a
+                number of channels, H and W are image height and width.
+            tokenize_newline_separately (`bool`, defaults to `True`):
+                Adds a separately tokenized '\n' at the end of the prompt.
+            padding (`bool`, `str` or [`~utils.PaddingStrategy`], *optional*, defaults to `False`):
+                Select a strategy to pad the returned sequences (according to the model's padding side and padding
+                index) among:
+                - `True` or `'longest'`: Pad to the longest sequence in the batch (or no padding if only a single
+                  sequence if provided).
+                - `'max_length'`: Pad to a maximum length specified with the argument `max_length` or to the maximum
+                  acceptable input length for the model if that argument is not provided.
+                - `False` or `'do_not_pad'` (default): No padding (i.e., can output a batch with sequences of different
+                  lengths).
+            max_length (`int`, *optional*):
+                Maximum length of the returned list and optionally padding length (see above).
+            truncation (`bool`, *optional*):
+                Activates truncation to cut input sequences longer than `max_length` to `max_length`.
+            return_tensors (`str` or [`~utils.TensorType`], *optional*):
+                If set, will return tensors of a particular framework. Acceptable values are:
+
+                - `'tf'`: Return TensorFlow `tf.constant` objects.
+                - `'pt'`: Return PyTorch `torch.Tensor` objects.
+                - `'np'`: Return NumPy `np.ndarray` objects.
+                - `'jax'`: Return JAX `jnp.ndarray` objects.
+
+        Returns:
+            [`BatchFeature`]: A [`BatchFeature`] with the following fields:
+
+            - **input_ids** -- List of token ids to be fed to a model. Returned when `text` is not `None`. If `suffix`
+              is provided, the `input_ids` will also contain the suffix input ids.
+            - **attention_mask** -- List of indices specifying which tokens should be attended to by the model (when
+              `return_attention_mask=True` or if *"attention_mask"* is in `self.model_input_names` and if `text` is not
+              `None`).
+            - **pixel_values** -- Pixel values to be fed to a model. Returned when `images` is not `None`.
+            - **labels** -- Labels compatible with training if `suffix` is not None
+        """
+
+        return_token_type_ids = False
+
+        if images is None:
+            raise ValueError("`images` are expected as arguments to a `Florence2Processor` instance.")
+        if text is None:
+            logger.warning_once(
+                "You are using Florence-2 without a text prompt."
+            )
+            text = ""
+
+        if isinstance(text, List) and isinstance(images, List):
+            if len(images) < len(text):
+                raise ValueError(
+                    f"Received {len(images)} images for {len(text)} prompts. Each prompt should be associated with an image."
+                )
+        if _is_str_or_image(text):
+            text = [text]
+        elif isinstance(text, list) and _is_str_or_image(text[0]):
+            pass
+
+        pixel_values = self.image_processor(
+            images,
+            size=size,
+            do_resize=do_resize,
+            do_normalize=do_normalize,
+            return_tensors=return_tensors,
+            image_mean=image_mean,
+            image_std=image_std,
+            input_data_format=input_data_format,
+            data_format=data_format,
+            resample=resample,
+            do_convert_rgb=do_convert_rgb,
+        )["pixel_values"]
+
+        if max_length is not None:
+            max_length -= self.image_seq_length  # max_length has to account for the image tokens
+
+        text = self._construct_prompts(text)
+
+        # inputs = self.tokenizer(
+        #     text,
+        #     return_tensors=return_tensors,
+        #     padding=padding,
+        #     max_length=max_length,
+        #     truncation=truncation,
+        #     return_token_type_ids=return_token_type_ids,
+        # )
+        print(text)
+        inputs = []
+        for t in text:
+            tokenized = self.tokenizer_image_token(t, self.tokenizer)
+            tokenized_tensor = torch.tensor(tokenized, dtype=torch.long)
+            inputs.append(tokenized_tensor)
+        inputs = torch.nn.utils.rnn.pad_sequence(inputs, batch_first=True, padding_value=self.tokenizer.pad_token_id)
+        inputs = {'input_ids': inputs}
+        # print(inputs)
+
+        return_data = {**inputs, "pixel_values": pixel_values}
+
+        if return_token_type_ids:
+            labels = inputs["input_ids"].masked_fill(inputs["token_type_ids"] == 0, -100)
+            return_data.update({"labels": labels})
+        return BatchFeature(data=return_data)
+
+    # Copied from transformers.models.clip.processing_clip.CLIPProcessor.batch_decode with CLIP->Florence2
+    def batch_decode(self, *args, **kwargs):
+        """
+        This method forwards all its arguments to BartTokenizerFast's [`~PreTrainedTokenizer.batch_decode`]. Please
+        refer to the docstring of this method for more information.
+        """
+        return self.tokenizer.batch_decode(*args, **kwargs)
+
+    # Copied from transformers.models.clip.processing_clip.CLIPProcessor.decode with CLIP->Florence2
+    def decode(self, *args, **kwargs):
+        """
+        This method forwards all its arguments to BartTokenizerFast's [`~PreTrainedTokenizer.decode`]. Please refer to
+        the docstring of this method for more information.
+        """
+        return self.tokenizer.decode(*args, **kwargs)
+
+    @property
+    # Copied from transformers.models.clip.processing_clip.CLIPProcessor.model_input_names with CLIP->Florence2
+    def model_input_names(self):
+        tokenizer_input_names = self.tokenizer.model_input_names
+        image_processor_input_names = self.image_processor.model_input_names
+        return list(dict.fromkeys(tokenizer_input_names + image_processor_input_names))
+
+    def post_process_generation(self, text, task, image_size):
+        """
+        Post-process the output of the model to each of the task outputs.
+
+        Args:
+            text (`str`): The text to post-process.
+            task (`str`): The task to post-process the text for.
+            image_size (`Tuple[int, int]`): The size of the image. height x width.
+        """
+
+        task_answer_post_processing_type = self.tasks_answer_post_processing_type.get(task, 'pure_text')
+        task_answer = self.post_processor(
+            text=text,
+            image_size=image_size,
+            parse_tasks=task_answer_post_processing_type,
+        )[task_answer_post_processing_type]
+
+        if task_answer_post_processing_type == 'pure_text':
+            final_answer = task_answer
+            # remove the special tokens
+            final_answer = final_answer.replace('<s>', '').replace('</s>', '\n')
+        elif task_answer_post_processing_type in ['od', 'description_with_bboxes', 'bboxes']:
+            od_instances = task_answer
+            bboxes_od = [_od_instance['bbox'] for _od_instance in od_instances]
+            labels_od = [str(_od_instance['cat_name']) for _od_instance in od_instances]
+            final_answer = {'bboxes': bboxes_od, 'labels': labels_od}
+        elif task_answer_post_processing_type in ['ocr']:
+            bboxes = [_od_instance['quad_box'] for _od_instance in task_answer]
+            labels = [str(_od_instance['text']) for _od_instance in task_answer]
+            final_answer = {'quad_boxes': bboxes, 'labels': labels}
+        elif task_answer_post_processing_type in ['phrase_grounding']:
+            bboxes = []
+            labels = []
+            for _grounded_phrase in task_answer:
+                for _bbox in _grounded_phrase['bbox']:
+                    bboxes.append(_bbox)
+                    labels.append(_grounded_phrase['cat_name'])
+            final_answer = {'bboxes': bboxes, 'labels': labels}
+        elif task_answer_post_processing_type in ['description_with_polygons', 'polygons']:
+            labels = []
+            polygons = []
+            for result in task_answer:
+                label = result['cat_name']
+                _polygons = result['polygons']
+                labels.append(label)
+                polygons.append(_polygons)
+            final_answer = {'polygons': polygons, 'labels': labels}
+        elif task_answer_post_processing_type in ['description_with_bboxes_or_polygons']:
+            bboxes = []
+            bboxes_labels = []
+            polygons = []
+            polygons_labels = []
+            for result in task_answer:
+                label = result['cat_name']
+                if 'polygons' in result:
+                    _polygons = result['polygons']
+                    polygons.append(_polygons)
+                    polygons_labels.append(label)
+                else:
+                    _bbox = result['bbox']
+                    bboxes.append(_bbox)
+                    bboxes_labels.append(label)
+            final_answer = {'bboxes': bboxes, 'bboxes_labels': bboxes_labels, 'polygons': polygons, 'polygons_labels': polygons_labels}
+        else:
+            raise ValueError('Unknown task answer post processing type: {}'.format(task_answer_post_processing_type))
+
+        final_answer = {
+            task: final_answer}
+        return final_answer 
+
+class BoxQuantizer(object):
+    def __init__(self, mode, bins):
+        self.mode = mode
+        self.bins = bins
+
+    def quantize(self, boxes: torch.Tensor, size):
+        bins_w, bins_h = self.bins  # Quantization bins.
+        size_w, size_h = size       # Original image size.
+        size_per_bin_w = size_w / bins_w
+        size_per_bin_h = size_h / bins_h
+        xmin, ymin, xmax, ymax = boxes.split(1, dim=-1)  # Shape: 4 * [N, 1].
+
+        if self.mode == 'floor':
+            quantized_xmin = (
+                xmin / size_per_bin_w).floor().clamp(0, bins_w - 1)
+            quantized_ymin = (
+                ymin / size_per_bin_h).floor().clamp(0, bins_h - 1)
+            quantized_xmax = (
+                xmax / size_per_bin_w).floor().clamp(0, bins_w - 1)
+            quantized_ymax = (
+                ymax / size_per_bin_h).floor().clamp(0, bins_h - 1)
+
+        elif self.mode == 'round':
+            raise NotImplementedError()
+
+        else:
+            raise ValueError('Incorrect quantization type.')
+
+        quantized_boxes = torch.cat(
+            (quantized_xmin, quantized_ymin, quantized_xmax, quantized_ymax), dim=-1
+        ).int()
+
+        return quantized_boxes
+
+    def dequantize(self, boxes: torch.Tensor, size):
+        bins_w, bins_h = self.bins  # Quantization bins.
+        size_w, size_h = size       # Original image size.
+        size_per_bin_w = size_w / bins_w
+        size_per_bin_h = size_h / bins_h
+        xmin, ymin, xmax, ymax = boxes.split(1, dim=-1)  # Shape: 4 * [N, 1].
+
+        if self.mode == 'floor':
+            # Add 0.5 to use the center position of the bin as the coordinate.
+            dequantized_xmin = (xmin + 0.5) * size_per_bin_w
+            dequantized_ymin = (ymin + 0.5) * size_per_bin_h
+            dequantized_xmax = (xmax + 0.5) * size_per_bin_w
+            dequantized_ymax = (ymax + 0.5) * size_per_bin_h
+
+        elif self.mode == 'round':
+            raise NotImplementedError()
+
+        else:
+            raise ValueError('Incorrect quantization type.')
+
+        dequantized_boxes = torch.cat(
+            (dequantized_xmin, dequantized_ymin,
+             dequantized_xmax, dequantized_ymax), dim=-1
+        )
+
+        return dequantized_boxes
+
+
+class CoordinatesQuantizer(object):
+    """
+    Quantize coornidates (Nx2)
+    """
+
+    def __init__(self, mode, bins):
+        self.mode = mode
+        self.bins = bins
+
+    def quantize(self, coordinates: torch.Tensor, size):
+        bins_w, bins_h = self.bins  # Quantization bins.
+        size_w, size_h = size       # Original image size.
+        size_per_bin_w = size_w / bins_w
+        size_per_bin_h = size_h / bins_h
+        assert coordinates.shape[-1] == 2, 'coordinates should be shape (N, 2)'
+        x, y = coordinates.split(1, dim=-1)  # Shape: 4 * [N, 1].
+
+        if self.mode == 'floor':
+            quantized_x = (x / size_per_bin_w).floor().clamp(0, bins_w - 1)
+            quantized_y = (y / size_per_bin_h).floor().clamp(0, bins_h - 1)
+
+        elif self.mode == 'round':
+            raise NotImplementedError()
+
+        else:
+            raise ValueError('Incorrect quantization type.')
+
+        quantized_coordinates = torch.cat(
+            (quantized_x, quantized_y), dim=-1
+        ).int()
+
+        return quantized_coordinates
+
+    def dequantize(self, coordinates: torch.Tensor, size):
+        bins_w, bins_h = self.bins  # Quantization bins.
+        size_w, size_h = size       # Original image size.
+        size_per_bin_w = size_w / bins_w
+        size_per_bin_h = size_h / bins_h
+        assert coordinates.shape[-1] == 2, 'coordinates should be shape (N, 2)'
+        x, y = coordinates.split(1, dim=-1)  # Shape: 4 * [N, 1].
+
+        if self.mode == 'floor':
+            # Add 0.5 to use the center position of the bin as the coordinate.
+            dequantized_x = (x + 0.5) * size_per_bin_w
+            dequantized_y = (y + 0.5) * size_per_bin_h
+
+        elif self.mode == 'round':
+            raise NotImplementedError()
+
+        else:
+            raise ValueError('Incorrect quantization type.')
+
+        dequantized_coordinates = torch.cat(
+            (dequantized_x, dequantized_y), dim=-1
+        )
+
+        return dequantized_coordinates
+
+
+class Florence2PostProcesser(object):
+    """
+    Florence-2 post process for converting text prediction to various tasks results. 
+
+    Args:
+        config: A dict of configs.
+        tokenizer: A tokenizer for decoding text to spans.
+        sample config:
+            UNIFIED_POST_PROCESS:
+                # commom configs
+                NUM_BBOX_HEIGHT_BINS: 1000
+                NUM_BBOX_WIDTH_BINS: 1000
+                COORDINATES_HEIGHT_BINS: 1000
+                COORDINATES_WIDTH_BINS: 1000
+                # task specific configs, override the common configs
+                PRASE_TASKS:
+                    - TASK_NAME: 'video_dense_caption'
+                      PATTERN: 'r<time_(\d+)><time_(\d+)>([a-zA-Z0-9 ]+)'
+                      SCORE_MODE: 'avg_cat_name_scores'
+                      NUM_BINS: 100
+                    - TASK_NAME: 'od'
+                      PATTERN: 'r<loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)>([a-zA-Z0-9 ]+)'
+                      SCORE_MODE: 'avg_cat_name_scores'
+
+    Returns:
+        parsed_dict (dict): A dict of parsed results.
+    """
+    def __init__(
+        self,
+        tokenizer=None
+    ):
+        parse_tasks = []
+        parse_task_configs = {}
+        config = self._create_default_config()
+        for task in config['PARSE_TASKS']:
+            parse_tasks.append(task['TASK_NAME'])
+            parse_task_configs[task['TASK_NAME']] = task
+
+        self.config = config
+        self.parse_tasks = parse_tasks
+        self.parse_tasks_configs = parse_task_configs
+
+        self.tokenizer =  tokenizer
+        if self.tokenizer is not None:
+            self.all_special_tokens = set(self.tokenizer.all_special_tokens)
+
+        self.init_quantizers()
+        self.black_list_of_phrase_grounding = self._create_black_list_of_phrase_grounding()
+
+    def _create_black_list_of_phrase_grounding(self):
+        black_list = {}
+
+        if 'phrase_grounding' in self.parse_tasks and self.parse_tasks_configs['phrase_grounding']['FILTER_BY_BLACK_LIST']:
+            black_list =  set(
+                ['it', 'I', 'me', 'mine',
+                 'you', 'your', 'yours',
+                 'he', 'him', 'his',
+                 'she', 'her', 'hers',
+                 'they', 'them', 'their', 'theirs',
+                 'one', 'oneself',
+                 'we', 'us', 'our', 'ours',
+                 'you', 'your', 'yours',
+                 'they', 'them', 'their', 'theirs',
+                 'mine', 'yours', 'his', 'hers', 'its',
+                 'ours', 'yours', 'theirs',
+                 'myself', 'yourself', 'himself', 'herself', 'itself',
+                 'ourselves', 'yourselves', 'themselves',
+                 'this', 'that',
+                 'these', 'those',
+                 'who', 'whom', 'whose', 'which', 'what',
+                 'who', 'whom', 'whose', 'which', 'that',
+                 'all', 'another', 'any', 'anybody', 'anyone', 'anything',
+                 'each', 'everybody', 'everyone', 'everything',
+                 'few', 'many', 'nobody', 'none', 'one', 'several',
+                 'some', 'somebody', 'someone', 'something',
+                 'each other', 'one another',
+                 'myself', 'yourself', 'himself', 'herself', 'itself',
+                 'ourselves', 'yourselves', 'themselves',
+                 'the image', 'image', 'images', 'the', 'a', 'an', 'a group',
+                 'other objects', 'lots', 'a set',
+                 ]
+            )
+
+        return black_list
+    
+    def _create_default_config(self):
+        config = {
+            'NUM_BBOX_HEIGHT_BINS': 1000,
+            'NUM_BBOX_WIDTH_BINS': 1000,
+            'BOX_QUANTIZATION_MODE': 'floor',
+            'COORDINATES_HEIGHT_BINS': 1000,
+            'COORDINATES_WIDTH_BINS': 1000,
+            'COORDINATES_QUANTIZATION_MODE': 'floor',
+            'PARSE_TASKS': [
+                {
+                    'TASK_NAME': 'od',
+                    'PATTERN': r'([a-zA-Z0-9 ]+)<loc_(\\d+)><loc_(\\d+)><loc_(\\d+)><loc_(\\d+)>'
+                },
+                {
+                    'TASK_NAME': 'ocr',
+                    'PATTERN':  r'(.+?)<loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)>',
+                    'AREA_THRESHOLD': 0.01
+                },
+                {
+                    'TASK_NAME': 'phrase_grounding',
+                    'FILTER_BY_BLACK_LIST': True
+                },
+                {
+                    'TASK_NAME': 'pure_text',
+                },
+                {
+                    'TASK_NAME': 'description_with_bboxes',
+                },
+                {
+                    'TASK_NAME': 'description_with_polygons',
+                },
+                {
+                    'TASK_NAME': 'polygons',
+                },
+                {
+                    'TASK_NAME': 'bboxes',
+                },
+                {
+                    'TASK_NAME': 'description_with_bboxes_or_polygons',
+                }
+            ]
+        }
+
+        return config
+
+    def init_quantizers(self):
+        # we have box_quantizer (od, grounding) and coordinates_quantizer (ocr, referring_segmentation)
+        num_bbox_height_bins = self.config.get('NUM_BBOX_HEIGHT_BINS', 1000)
+        num_bbox_width_bins = self.config.get('NUM_BBOX_WIDTH_BINS', 1000)
+        box_quantization_mode = self.config.get('BOX_QUANTIZATION_MODE', 'floor')
+        self.box_quantizer = BoxQuantizer(
+            box_quantization_mode,
+            (num_bbox_width_bins, num_bbox_height_bins),
+        )
+        
+        num_bbox_height_bins = self.config['COORDINATES_HEIGHT_BINS'] if 'COORDINATES_HEIGHT_BINS' in self.config else self.config.get('NUM_BBOX_HEIGHT_BINS', 1000)
+        num_bbox_width_bins = self.config['COORDINATES_WIDTH_BINS'] if 'COORDINATES_WIDTH_BINS' in self.config else self.config.get('NUM_BBOX_WIDTH_BINS', 1000)
+        box_quantization_mode = self.config.get('COORDINATES_QUANTIZATION_MODE') if 'COORDINATES_QUANTIZATION_MODE' in self.config else self.config.get('BOX_QUANTIZATION_MODE', 'floor')
+        self.coordinates_quantizer = CoordinatesQuantizer(
+            box_quantization_mode,
+            (num_bbox_width_bins, num_bbox_height_bins),
+        )
+
+    def decode_with_spans(self, tokenizer, token_ids):
+        filtered_tokens = tokenizer.convert_ids_to_tokens(
+            token_ids, skip_special_tokens=False)
+        assert len(filtered_tokens) == len(token_ids)
+
+        # To avoid mixing byte-level and unicode for byte-level BPT
+        # we need to build string separately for added tokens and byte-level tokens
+        # cf. https://github.com/huggingface/transformers/issues/1133
+        sub_texts = []
+        for token in filtered_tokens:
+            if token in self.all_special_tokens:
+                sub_texts.append(token)
+            else:
+                if isinstance(tokenizer, (BartTokenizer, BartTokenizerFast)):
+                    sub_text = tokenizer.convert_tokens_to_string([token])
+                elif isinstance(tokenizer, (T5Tokenizer, T5TokenizerFast)):
+                    # Ref: https://github.com/google/sentencepiece#whitespace-is-treated-as-a-basic-symbol
+                    # Note: Do not strip sub_text as it may have functional whitespace
+                    sub_text = token.replace('▁', ' ')
+                else:
+                    raise ValueError(f'type {type(tokenizer)} not supported')
+                sub_texts.append(sub_text)
+
+        text = ''
+        spans = []
+        for sub_text in sub_texts:
+            span = (len(text), len(text) + len(sub_text))  # [start index, end index).
+            text += sub_text
+            spans.append(span)
+
+        # Text format:
+        # 1. T5Tokenizer/T5TokenizerFast: 
+        #      "<loc_1><loc_2><loc_3><loc_4> transplanting dog<loc_1><loc_2><loc_3><loc_4> cat</s>"
+        #    Equivalent to t5_tokenizer.decode(input_ids, skip_special_tokens=False, clean_up_tokenization_spaces=False, spaces_between_special_tokens=False)
+        # 2. BartTokenizer (need to double check):
+        #      "<s><loc_1><loc_2><loc_3><loc_4>transplanting dog<loc_1><loc_2><loc_3><loc_4>cat</s>"
+        #    Equivalent to bart_tokenizer.decode(input_ids, skip_special_tokens=False, clean_up_tokenization_spaces=False, spaces_between_special_tokens=False)
+        return text, spans
+
+    def parse_od_from_text_and_spans(
+        self,
+        text,
+        pattern,
+        image_size,
+        phrase_centric=False
+    ):
+        parsed = list(re.finditer(pattern, text))
+
+        instances = []
+        for i in range(len(parsed)):
+            # Prepare instance.
+            instance = {}
+
+            if phrase_centric:
+                bbox_bins = [int(parsed[i].group(j)) for j in range(2, 6)]
+            else:
+                bbox_bins = [int(parsed[i].group(j)) for j in range(1, 5)]
+            instance['bbox'] = self.box_quantizer.dequantize(
+                boxes=torch.tensor(bbox_bins),
+                size=image_size
+            ).tolist()  
+
+            if phrase_centric:
+                instance['cat_name'] = parsed[i].group(1).lower().strip()
+            else:
+                instance['cat_name'] = parsed[i].group(5).lower().strip()
+            instances.append(instance)
+
+        return instances
+
+    def parse_ocr_from_text_and_spans(self, 
+                                    text, 
+                                     pattern, 
+                                     image_size,
+                                     area_threshold=-1.0,
+        ):
+        bboxes = []
+        labels = []
+        text = text.replace('<s>', '')
+        # ocr with regions
+        parsed = re.findall(pattern, text)
+        instances = []
+        image_width, image_height = image_size
+
+        for ocr_line in parsed:
+            ocr_content = ocr_line[0]
+            quad_box = ocr_line[1:]
+            quad_box = [int(i) for i in quad_box]
+            quad_box = self.coordinates_quantizer.dequantize(
+                torch.tensor(np.array(quad_box).reshape(-1, 2)),
+                size=image_size
+            ).reshape(-1).tolist()
+
+            if area_threshold > 0:
+                x_coords = [i for i in quad_box[0::2]]
+                y_coords = [i for i in quad_box[1::2]]
+
+                # apply the Shoelace formula
+                area = 0.5 * abs(sum(x_coords[i] * y_coords[i + 1] - x_coords[i + 1] * y_coords[i] for i in range(4 - 1)))
+
+                if area < (image_width * image_height) * area_threshold:
+                    continue
+
+            bboxes.append(quad_box)
+            labels.append(ocr_content)
+            instances.append({
+                'quad_box': quad_box,
+                'text': ocr_content,
+            })
+        return instances
+
+    def parse_phrase_grounding_from_text_and_spans(self, text, pattern, image_size):
+        # ignore <s> </s> and <pad>
+        cur_span = 0
+        if text.startswith('<s>'):   
+            cur_span += 3
+
+        text = text.replace('<s>', '')
+        text = text.replace('</s>', '')
+        text = text.replace('<pad>', '')
+
+        pattern = r"([^<]+(?:<loc_\d+>){4,})"
+        phrases = re.findall(pattern, text)
+    
+        # pattern should be text pattern and od pattern
+        pattern = r'^\s*(.*?)(?=<od>|</od>|<box>|</box>|<bbox>|</bbox>|<loc_)'
+        box_pattern = r'<loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)>'
+
+        instances = []
+        for pharse_text in phrases:
+            phrase_text_strip = pharse_text.replace('<ground>', '', 1)
+            phrase_text_strip = pharse_text.replace('<obj>', '', 1)
+
+            if phrase_text_strip == '':
+                cur_span += len(pharse_text)
+                continue
+
+            # Prepare instance.
+            instance = {}
+
+            # parse phrase, get string 
+            phrase = re.search(pattern, phrase_text_strip)
+            if phrase is None:
+                cur_span += len(pharse_text)
+                continue
+
+            # parse bboxes by box_pattern
+            bboxes_parsed = list(re.finditer(box_pattern, pharse_text))
+            if len(bboxes_parsed) == 0:
+                cur_span += len(pharse_text)
+                continue
+
+            phrase = phrase.group()
+            # remove leading and trailing spaces
+            phrase = phrase.strip()
+
+            if phrase in self.black_list_of_phrase_grounding:
+                cur_span += len(pharse_text)
+                continue
+
+            # a list of list 
+            bbox_bins = [[int(_bboxes_parsed.group(j)) for j in range(1, 5)] for _bboxes_parsed in bboxes_parsed]
+            instance['bbox'] = self.box_quantizer.dequantize(
+                boxes=torch.tensor(bbox_bins),
+                size=image_size
+            ).tolist()  
+
+            # exclude non-ascii characters
+            phrase = phrase.encode('ascii',errors='ignore').decode('ascii')
+            instance['cat_name'] = phrase
+
+            instances.append(instance)
+
+        return instances
+
+    def parse_description_with_bboxes_from_text_and_spans(self, text, pattern, image_size, allow_empty_phrase=False):
+        # temporary parse solution, split by '.'
+        # ignore <s> </s> and <pad>
+
+        text = text.replace('<s>', '')
+        text = text.replace('</s>', '')
+        text = text.replace('<pad>', '')
+
+        if allow_empty_phrase:
+            pattern = rf"(?:(?:<loc_\d+>){{4,}})"
+        else:
+            pattern = r"([^<]+(?:<loc_\d+>){4,})"
+        phrases = re.findall(pattern, text)
+    
+        # pattern should be text pattern and od pattern
+        pattern = r'^\s*(.*?)(?=<od>|</od>|<box>|</box>|<bbox>|</bbox>|<loc_)'
+        box_pattern = r'<loc_(\d+)><loc_(\d+)><loc_(\d+)><loc_(\d+)>'
+
+        instances = []
+        for pharse_text in phrases:
+            phrase_text_strip = pharse_text.replace('<ground>', '', 1)
+            phrase_text_strip = pharse_text.replace('<obj>', '', 1)
+
+            if phrase_text_strip == '' and not allow_empty_phrase:
+                continue
+
+            # parse phrase, get string 
+            phrase = re.search(pattern, phrase_text_strip)
+            if phrase is None:
+                continue
+
+            phrase = phrase.group()
+            # remove leading and trailing spaces
+            phrase = phrase.strip()
+
+            # parse bboxes by box_pattern
+            bboxes_parsed = list(re.finditer(box_pattern, pharse_text))
+            if len(bboxes_parsed) == 0:
+                continue
+
+            # a list of list 
+            bbox_bins = [[int(_bboxes_parsed.group(j)) for j in range(1, 5)] for _bboxes_parsed in bboxes_parsed]
+
+            bboxes = self.box_quantizer.dequantize(
+                boxes=torch.tensor(bbox_bins),
+                size=image_size
+            ).tolist()  
+
+            phrase = phrase.encode('ascii',errors='ignore').decode('ascii')
+            for _bboxes in bboxes:
+                # Prepare instance.
+                instance = {}
+                instance['bbox'] = _bboxes
+                # exclude non-ascii characters
+                instance['cat_name'] = phrase
+                instances.append(instance)
+
+        return instances
+
+    def parse_description_with_polygons_from_text_and_spans(self, text, pattern, image_size, 
+                                                            allow_empty_phrase=False,
+                                                            polygon_sep_token='<sep>',
+                                                            polygon_start_token='<poly>',
+                                                            polygon_end_token='</poly>',
+                                                            with_box_at_start=False,
+                                                            ):
+        
+        # ref_seg format: '<expression><x1><y1><x2><y2><><><sep><><><><>'
+        # ignore <s> </s> and <pad>
+
+        text = text.replace('<s>', '')
+        text = text.replace('</s>', '')
+        text = text.replace('<pad>', '')
+
+        if allow_empty_phrase:
+            pattern = rf"(?:(?:<loc_\d+>|{re.escape(polygon_sep_token)}|{re.escape(polygon_start_token)}|{re.escape(polygon_end_token)}){{4,}})"
+        else:
+            # [^<]+: This part matches one or more characters that are not the < symbol. 
+            # The ^ inside the square brackets [] is a negation, meaning it matches anything except <.
+            #
+            pattern = rf"([^<]+(?:<loc_\d+>|{re.escape(polygon_sep_token)}|{re.escape(polygon_start_token)}|{re.escape(polygon_end_token)}){{4,}})"
+        phrases = re.findall(pattern, text)
+
+        phrase_string_pattern = r'^\s*(.*?)(?=<od>|</od>|<box>|</box>|<bbox>|</bbox>|<loc_|<poly>)'
+        box_pattern =  rf'((?:<loc_\d+>)+)(?:{re.escape(polygon_sep_token)}|$)'
+
+        # one polygons instance is separated by polygon_start_token and polygon_end_token
+        polygons_instance_pattern = rf'{re.escape(polygon_start_token)}(.*?){re.escape(polygon_end_token)}'
+    
+        instances = []
+        for phrase_text in phrases:
+
+            # exclude loc_\d+>
+            # need to get span if want to include category score
+            phrase_text_strip = re.sub(r'^loc_\d+>', '', phrase_text, count=1)
+
+            # phrase = phrase.replace('<poly>', '')
+            # phrase = phrase.replace('poly>', '')
+
+            if phrase_text_strip == '' and not allow_empty_phrase:
+                continue
+
+
+            # parse phrase, get string 
+            phrase = re.search(phrase_string_pattern, phrase_text_strip)
+            if phrase is None:
+                continue
+            phrase = phrase.group()
+            # remove leading and trailing spaces
+            phrase = phrase.strip()
+
+            # parse bboxes by box_pattern
+
+            # split by polygon_start_token and polygon_end_token first using polygons_instance_pattern
+            if polygon_start_token in phrase_text and polygon_end_token in phrase_text:
+                polygons_instances_parsed = list(re.finditer(polygons_instance_pattern, phrase_text))
+            else:
+                polygons_instances_parsed = [phrase_text]
+
+            for _polygons_instances_parsed in polygons_instances_parsed:
+                # Prepare instance.
+                instance = {}
+
+                # polygons_parsed= list(re.finditer(box_pattern, phrase_text))
+                if isinstance(_polygons_instances_parsed, str): 
+                    polygons_parsed= list(re.finditer(box_pattern, _polygons_instances_parsed))
+                else:
+                    polygons_parsed= list(re.finditer(box_pattern, _polygons_instances_parsed.group(1)))
+                if len(polygons_parsed) == 0:
+                    continue
+
+                # a list of list (polygon)
+                bbox = []
+                polygons = []
+                for _polygon_parsed in polygons_parsed:
+                    # group 1: whole <loc_\d+>...</loc_\d+>
+                    _polygon = _polygon_parsed.group(1)
+                    # parse into list of int
+                    _polygon = [int(_loc_parsed.group(1)) for _loc_parsed in re.finditer(r'<loc_(\d+)>', _polygon)]
+                    if with_box_at_start and len(bbox) == 0:
+                        if len(_polygon) > 4:
+                            # no valid bbox prediction
+                            bbox = _polygon[:4]
+                            _polygon = _polygon[4:]
+                        else:
+                            bbox = [0, 0, 0, 0]
+                    # abandon last element if is not paired 
+                    if len(_polygon) % 2 == 1:
+                        _polygon = _polygon[:-1]
+                    
+                    # reshape into (n, 2)
+                    _polygon = self.coordinates_quantizer.dequantize(
+                        torch.tensor(np.array(_polygon).reshape(-1, 2)),
+                        size=image_size
+                    ).reshape(-1).tolist()
+                    # reshape back
+                    polygons.append(_polygon)
+
+                instance['cat_name'] = phrase
+                instance['polygons'] = polygons
+                if len(bbox) != 0:
+                    instance['bbox'] = self.box_quantizer.dequantize(
+                        boxes=torch.tensor([bbox]),
+                        size=image_size
+                    ).tolist()[0]  
+
+                instances.append(instance)
+
+        return instances
+
+    def __call__(
+        self,
+        text=None,
+        image_size=None,
+        parse_tasks=None,
+    ):
+        """
+        Args:
+            text: model outputs
+            image_size: (width, height)
+            parse_tasks: a list of tasks to parse, if None, parse all tasks.
+
+        """
+        if parse_tasks is not None:
+            if isinstance(parse_tasks, str):
+                parse_tasks = [parse_tasks]
+            for _parse_task in parse_tasks:
+                assert _parse_task in self.parse_tasks, f'parse task {_parse_task} not supported'
+        
+        # sequence or text should be provided 
+        assert text is not None, 'text should be provided'
+
+        parsed_dict = {
+            'text': text
+        }
+
+        for task in self.parse_tasks:
+            if parse_tasks is not None and task not in parse_tasks:
+                continue
+
+            pattern = self.parse_tasks_configs[task].get('PATTERN', None)
+
+            if task == 'ocr':
+                instances = self.parse_ocr_from_text_and_spans(
+                    text,
+                    pattern=pattern,
+                    image_size=image_size,
+                    area_threshold=self.parse_tasks_configs[task].get('AREA_THRESHOLD', 0.01),
+                )
+                parsed_dict['ocr'] = instances
+            elif task == 'phrase_grounding':
+                instances = self.parse_phrase_grounding_from_text_and_spans( 
+                    text,
+                    pattern=pattern,
+                    image_size=image_size,
+                )
+                parsed_dict['phrase_grounding'] = instances
+            elif task == 'pure_text':
+                parsed_dict['pure_text'] = text 
+            elif task == 'description_with_bboxes':
+                instances = self.parse_description_with_bboxes_from_text_and_spans( 
+                    text,
+                    pattern=pattern,
+                    image_size=image_size,
+                )
+                parsed_dict['description_with_bboxes'] = instances
+            elif task == 'description_with_polygons':
+                instances = self.parse_description_with_polygons_from_text_and_spans( 
+                    text,
+                    pattern=pattern,
+                    image_size=image_size,
+                )
+                parsed_dict['description_with_polygons'] = instances
+            elif task == 'polygons':
+                instances = self.parse_description_with_polygons_from_text_and_spans( 
+                    text,
+                    pattern=pattern,
+                    image_size=image_size,
+                    allow_empty_phrase=True,
+                )
+                parsed_dict['polygons'] = instances
+            elif task == 'bboxes':
+                instances = self.parse_description_with_bboxes_from_text_and_spans( 
+                    text,
+                    pattern=pattern,
+                    image_size=image_size,
+                    allow_empty_phrase=True,
+                )
+                parsed_dict['bboxes'] = instances
+            elif task == 'description_with_bboxes_or_polygons':
+                if '<poly>' in text:
+                    # only support either polygons or bboxes, not both at the same time
+                    instances = self.parse_description_with_polygons_from_text_and_spans( 
+                        text,
+                        pattern=pattern,
+                        image_size=image_size,
+                    )
+                else:
+                    instances = self.parse_description_with_bboxes_from_text_and_spans( 
+                        text,
+                        pattern=pattern,
+                        image_size=image_size,
+                    )
+                parsed_dict['description_with_bboxes_or_polygons'] = instances
+            else:
+                raise ValueError("task {} is not supported".format(task))
+
+        return parsed_dict

tokenizer_config.json

@@ -0,0 +1,207 @@
+{
+  "add_bos_token": false,
+  "add_prefix_space": false,
+  "added_tokens_decoder": {
+    "151643": {
+      "content": "<|endoftext|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151644": {
+      "content": "<|im_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151645": {
+      "content": "<|im_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151646": {
+      "content": "<|object_ref_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151647": {
+      "content": "<|object_ref_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151648": {
+      "content": "<|box_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151649": {
+      "content": "<|box_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151650": {
+      "content": "<|quad_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151651": {
+      "content": "<|quad_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151652": {
+      "content": "<|vision_start|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151653": {
+      "content": "<|vision_end|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151654": {
+      "content": "<|vision_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151655": {
+      "content": "<|image_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151656": {
+      "content": "<|video_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "151657": {
+      "content": "<tool_call>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151658": {
+      "content": "</tool_call>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151659": {
+      "content": "<|fim_prefix|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151660": {
+      "content": "<|fim_middle|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151661": {
+      "content": "<|fim_suffix|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151662": {
+      "content": "<|fim_pad|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151663": {
+      "content": "<|repo_name|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    },
+    "151664": {
+      "content": "<|file_sep|>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": false
+    }
+  },
+  "additional_special_tokens": [
+    "<|im_start|>",
+    "<|im_end|>",
+    "<|object_ref_start|>",
+    "<|object_ref_end|>",
+    "<|box_start|>",
+    "<|box_end|>",
+    "<|quad_start|>",
+    "<|quad_end|>",
+    "<|vision_start|>",
+    "<|vision_end|>",
+    "<|vision_pad|>",
+    "<|image_pad|>",
+    "<|video_pad|>"
+  ],
+  "bos_token": null,
+  "chat_template": "{%- if tools %}\n    {{- '<|im_start|>system\\n' }}\n    {%- if messages[0]['role'] == 'system' %}\n        {{- messages[0]['content'] }}\n    {%- else %}\n        {{- 'You are a helpful assistant.' }}\n    {%- endif %}\n    {{- \"\\n\\n# Tools\\n\\nYou may call one or more functions to assist with the user query.\\n\\nYou are provided with function signatures within <tools></tools> XML tags:\\n<tools>\" }}\n    {%- for tool in tools %}\n        {{- \"\\n\" }}\n        {{- tool | tojson }}\n    {%- endfor %}\n    {{- \"\\n</tools>\\n\\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\\n<tool_call>\\n{\\\"name\\\": <function-name>, \\\"arguments\\\": <args-json-object>}\\n</tool_call><|im_end|>\\n\" }}\n{%- else %}\n    {%- if messages[0]['role'] == 'system' %}\n        {{- '<|im_start|>system\\n' + messages[0]['content'] + '<|im_end|>\\n' }}\n    {%- else %}\n        {{- '<|im_start|>system\\nYou are a helpful assistant.<|im_end|>\\n' }}\n    {%- endif %}\n{%- endif %}\n{%- for message in messages %}\n    {%- if (message.role == \"user\") or (message.role == \"system\" and not loop.first) or (message.role == \"assistant\" and not message.tool_calls) %}\n        {{- '<|im_start|>' + message.role + '\\n' + message.content + '<|im_end|>' + '\\n' }}\n    {%- elif message.role == \"assistant\" %}\n        {{- '<|im_start|>' + message.role }}\n        {%- if message.content %}\n            {{- '\\n' + message.content }}\n        {%- endif %}\n        {%- for tool_call in message.tool_calls %}\n            {%- if tool_call.function is defined %}\n                {%- set tool_call = tool_call.function %}\n            {%- endif %}\n            {{- '\\n<tool_call>\\n{\"name\": \"' }}\n            {{- tool_call.name }}\n            {{- '\", \"arguments\": ' }}\n            {{- tool_call.arguments | tojson }}\n            {{- '}\\n</tool_call>' }}\n        {%- endfor %}\n        {{- '<|im_end|>\\n' }}\n    {%- elif message.role == \"tool\" %}\n        {%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != \"tool\") %}\n            {{- '<|im_start|>user' }}\n        {%- endif %}\n        {{- '\\n<tool_response>\\n' }}\n        {{- message.content }}\n        {{- '\\n</tool_response>' }}\n        {%- if loop.last or (messages[loop.index0 + 1].role != \"tool\") %}\n            {{- '<|im_end|>\\n' }}\n        {%- endif %}\n    {%- endif %}\n{%- endfor %}\n{%- if add_generation_prompt %}\n    {{- '<|im_start|>assistant\\n' }}\n{%- endif %}\n",
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "<|endoftext|>",
+  "errors": "replace",
+  "model_max_length": 131072,
+  "pad_token": "<|endoftext|>",
+  "split_special_tokens": false,
+  "tokenizer_class": "Qwen2Tokenizer",
+  "unk_token": null
+}