nexaai 1.0.19rc7__cp310-cp310-macosx_14_0_universal2.whl → 1.0.19rc8__cp310-cp310-macosx_14_0_universal2.whl

nexaai/binds/nexa_mlx/py-lib/vlm/modeling/models/qwen2_5_vl/qwen2_5_vl.py

@@ -1,168 +0,0 @@
-import glob
-import json
-from pathlib import Path
-from typing import Optional
-
-import mlx.core as mx
-import mlx.nn as nn
-from huggingface_hub import snapshot_download
-
-from .config import ModelConfig, TextConfig, VisionConfig
-from .language import LanguageModel
-from .vision import VisionModel
-
-
-class Model(nn.Module):
-    def __init__(self, config: ModelConfig):
-        super().__init__()
-        self.config = config
-        self.vision_tower = VisionModel(config.vision_config)
-        self.language_model = LanguageModel(config.text_config, config)
-
-    def get_input_embeddings(
-        self,
-        input_ids: Optional[mx.array] = None,
-        pixel_values: Optional[mx.array] = None,
-        image_grid_thw: Optional[mx.array] = None,
-    ):
-        if pixel_values is None:
-            return self.language_model.model.embed_tokens(input_ids)
-
-        dtype = self.vision_tower.patch_embed.proj.weight.dtype
-        pixel_values = pixel_values.astype(dtype)
-
-        # Get the input embeddings from the language model
-        inputs_embeds = self.language_model.model.embed_tokens(input_ids)
-
-        # Get the ouptut hidden states from the vision model
-        hidden_states = self.vision_tower(
-            pixel_values, image_grid_thw, output_hidden_states=False
-        )
-
-        # Insert special image tokens in the input_ids
-        final_inputs_embeds = self.merge_input_ids_with_image_features(
-            self.config.image_token_id,
-            self.config.video_token_id,
-            hidden_states,
-            inputs_embeds,
-            input_ids,
-        )
-        return final_inputs_embeds
-
-    @staticmethod
-    def merge_input_ids_with_image_features(
-        image_token_id,
-        video_token_id,
-        image_features,
-        inputs_embeds,
-        input_ids,
-    ):
-        """Merge image features into input embeddings at image token positions.
-
-        Args:
-            image_token_id: The token ID for image placeholders
-            video_token_id: The token ID for video placeholders (fallback)
-            image_features: Vision features from the vision tower [num_features, hidden_dim]
-            inputs_embeds: Input embeddings [batch_size, seq_len, hidden_dim]
-            input_ids: Input token IDs [batch_size, seq_len]
-            grid_thw: Grid dimensions for each image (optional, not used in simple case)
-
-        Returns:
-            Updated input embeddings with image features inserted
-        """
-        # Find positions of image tokens
-        image_positions = input_ids == image_token_id
-        if mx.sum(image_positions) == 0:
-            image_positions = input_ids == video_token_id
-
-        # Get dimensions
-        batch_size, seq_len = input_ids.shape
-
-        # Process each batch item
-        batch_outputs = []
-        feature_start_idx = 0
-
-        for batch_idx in range(batch_size):
-            # Get mask for this batch
-            image_mask = image_positions[batch_idx]
-            num_positions = mx.sum(image_mask).item()
-
-            if num_positions > 0:
-                # Extract features for this batch
-                batch_features = image_features[
-                    feature_start_idx : feature_start_idx + num_positions
-                ]
-
-                # Validate we have the right number of features
-                if batch_features.shape[0] != num_positions:
-                    raise ValueError(
-                        f"Number of image token positions ({num_positions}) does not match "
-                        f"number of image features ({batch_features.shape[0]}) for batch {batch_idx}"
-                    )
-
-                # Create indices for gathering
-                cumsum = mx.cumsum(image_mask.astype(mx.int32))
-                feature_indices = mx.where(image_mask, cumsum - 1, 0)
-
-                # Gather features
-                gathered_features = batch_features[feature_indices]
-
-                # Combine with original embeddings
-                image_mask_expanded = mx.expand_dims(image_mask, axis=-1)
-                batch_output = mx.where(
-                    image_mask_expanded, gathered_features, inputs_embeds[batch_idx]
-                )
-
-                feature_start_idx += num_positions
-            else:
-                # No image tokens in this batch item
-                batch_output = inputs_embeds[batch_idx]
-
-            batch_outputs.append(batch_output)
-
-        # Stack all batch outputs
-        return mx.stack(batch_outputs, axis=0)
-
-    @property
-    def layers(self):
-        return self.language_model.model.layers
-
-    def __call__(
-        self,
-        input_ids: mx.array,
-        pixel_values: Optional[mx.array] = None,
-        mask: Optional[mx.array] = None,
-        cache=None,
-        **kwargs,
-    ):
-        image_grid_thw = kwargs.pop("image_grid_thw", None)
-        video_grid_thw = kwargs.pop("video_grid_thw", None)
-        grid_thw = image_grid_thw if image_grid_thw is not None else video_grid_thw
-
-        inputs_embeds = self.get_input_embeddings(input_ids, pixel_values, grid_thw)
-
-        kwargs = {
-            "pixel_values": pixel_values,
-            "image_grid_thw": image_grid_thw,
-            "video_grid_thw": video_grid_thw,
-            **kwargs,
-        }
-
-        logits = self.language_model(
-            input_ids, inputs_embeds, mask=mask, cache=cache, **kwargs
-        )
-
-        return logits
-
-    def sanitize(self, weights):
-        def transform_key(key):
-            if "vision_tower" not in key:
-                key = key.replace("visual", "vision_tower")
-            if "language_model" not in key:
-                if "model" in key:
-                    key = key.replace("model", "language_model.model")
-                elif "lm_head" in key:
-                    key = key.replace("lm_head", "language_model.lm_head")
-            return key
-
-        return {transform_key(k): v for k, v in weights.items()}

nexaai/binds/nexa_mlx/py-lib/vlm/modeling/models/qwen2_5_vl/vision.py

@@ -1,414 +0,0 @@
-from typing import Optional
-
-import mlx.core as mx
-import mlx.nn as nn
-import numpy as np
-
-from .config import VisionConfig
-
-
-def check_array_shape(arr):
-    shape = arr.shape
-
-    # Check if the shape has 4 dimensions
-    if len(shape) not in [4, 5]:
-        return False
-
-    B, out_channels, kH, KW, t = shape
-
-    if t == 3:
-        return True
-
-    # Check if out_channels is the largest, and kH and KW are the same
-    if (out_channels >= kH) and (out_channels >= KW) and (kH == KW):
-        return True
-    else:
-        return False
-
-
-def rotate_half(x):
-    """Rotates half the hidden dims of the input."""
-    x1 = x[..., : x.shape[-1] // 2]
-    x2 = x[..., x.shape[-1] // 2 :]
-    return mx.concatenate([-x2, x1], axis=-1)
-
-
-def apply_rotary_pos_emb_vision(tensor, freqs) -> mx.array:
-    orig_dtype = tensor.dtype
-
-    cos = mx.cos(freqs)
-    sin = mx.sin(freqs)
-
-    cos = mx.expand_dims(cos, axis=1)  # Equivalent to unsqueeze(1)
-    cos = mx.tile(cos, (1, 1, 2))  # Equivalent to repeat(1, 1, 2)
-    cos = mx.expand_dims(cos, axis=0)  # Equivalent to [None, ...]
-
-    sin = mx.expand_dims(sin, axis=1)  # Equivalent to unsqueeze(1)
-    sin = mx.tile(sin, (1, 1, 2))  # Equivalent to repeat(1, 1, 2)
-    sin = mx.expand_dims(sin, axis=0)  # Equivalent to [None, ...]
-
-    output = (tensor * cos) + (rotate_half(tensor) * sin)
-    return output.astype(orig_dtype)
-
-
-class VisionRotaryEmbedding(nn.Module):
-    def __init__(self, dim: int, theta: float = 10000.0) -> None:
-        super().__init__()
-        self.dim = dim
-        self.theta = theta
-
-    def __call__(self, seqlen: int) -> mx.array:
-        inv_freq = 1.0 / (
-            self.theta ** (mx.arange(0, self.dim, 2, dtype=mx.float32) / self.dim)
-        )
-        seq = mx.arange(seqlen.item(), dtype=inv_freq.dtype)
-        freqs = mx.outer(seq, inv_freq)
-        return freqs
-
-
-class PatchEmbed(nn.Module):
-    def __init__(
-        self,
-        patch_size: int = 14,
-        temporal_patch_size: int = 2,
-        in_channels: int = 3,
-        hidden_size: int = 1152,
-    ) -> None:
-        super().__init__()
-        self.patch_size = patch_size
-        self.temporal_patch_size = temporal_patch_size
-        self.in_channels = in_channels
-        self.hidden_size = hidden_size
-
-        kernel_size = [temporal_patch_size, patch_size, patch_size]
-        self.proj = nn.Conv3d(
-            in_channels,
-            hidden_size,
-            kernel_size=kernel_size,
-            stride=kernel_size,
-            bias=False,
-        )
-
-    def __call__(self, hidden_states: mx.array) -> mx.array:
-        hidden_states = hidden_states.reshape(
-            -1,
-            self.in_channels,
-            self.temporal_patch_size,
-            self.patch_size,
-            self.patch_size,
-        ).moveaxis(1, 4)
-
-        hidden_states = self.proj(hidden_states)
-        hidden_states = hidden_states.reshape(-1, self.hidden_size)
-        return hidden_states
-
-
-class PatchMerger(nn.Module):
-    def __init__(self, dim: int, context_dim: int, spatial_merge_size: int = 2) -> None:
-        super().__init__()
-        self.hidden_size = context_dim * (spatial_merge_size**2)
-        self.ln_q = nn.RMSNorm(context_dim, eps=1e-6)
-        self.mlp = [
-            nn.Linear(self.hidden_size, self.hidden_size),
-            nn.GELU(),
-            nn.Linear(self.hidden_size, dim),
-        ]
-
-    def __call__(self, x: mx.array) -> mx.array:
-        x = self.ln_q(x).reshape(-1, self.hidden_size)
-        for layer in self.mlp:
-            x = layer(x)
-        return x
-
-
-class Attention(nn.Module):
-    def __init__(self, dim: int, num_heads: int = 16) -> None:
-        super().__init__()
-        self.num_heads = num_heads
-        self.head_dim = head_dim = dim // num_heads
-        self.scale = head_dim**-0.5
-        self.qkv = nn.Linear(dim, dim * 3, bias=True)
-        self.proj = nn.Linear(dim, dim)
-
-    def __call__(
-        self, x: mx.array, cu_seqlens: mx.array, rotary_pos_emb: mx.array = None
-    ) -> mx.array:
-        seq_length = x.shape[0]
-        qkv = (
-            self.qkv(x).reshape(seq_length, 3, self.num_heads, -1).transpose(1, 0, 2, 3)
-        )
-        q, k, v = mx.split(qkv, 3)
-
-        q = apply_rotary_pos_emb_vision(mx.expand_dims(q, 0), rotary_pos_emb)[0]
-        k = apply_rotary_pos_emb_vision(mx.expand_dims(k, 0), rotary_pos_emb)[0]
-        attention_mask = mx.full(
-            (1, seq_length, seq_length), mx.finfo(q.dtype).min, dtype=q.dtype
-        )
-
-        for i in range(1, len(cu_seqlens)):
-            start = int(cu_seqlens[i - 1])
-            end = int(cu_seqlens[i])
-            attention_mask[..., start:end, start:end] = 0
-
-        q = q.transpose(0, 2, 1, 3)
-        k = k.transpose(0, 2, 1, 3)
-        v = v.transpose(0, 2, 1, 3)
-
-        output = mx.fast.scaled_dot_product_attention(
-            q, k, v, scale=self.scale, mask=attention_mask
-        )
-        output = output.transpose(0, 2, 1, 3)
-        output = output.reshape(seq_length, -1)
-        return self.proj(output)
-
-
-class MLP(nn.Module):
-    def __init__(self, dim, hidden_dim):
-        super().__init__()
-        self.gate_proj = nn.Linear(dim, hidden_dim)
-        self.up_proj = nn.Linear(dim, hidden_dim)
-        self.down_proj = nn.Linear(hidden_dim, dim)
-
-    def __call__(self, x: mx.array) -> mx.array:
-        return self.down_proj(nn.silu(self.gate_proj(x)) * self.up_proj(x))
-
-
-class Qwen2VLVisionBlock(nn.Module):
-    def __init__(self, config: VisionConfig) -> None:
-        super().__init__()
-        self.norm1 = nn.RMSNorm(config.hidden_size, eps=1e-6)
-        self.norm2 = nn.RMSNorm(config.hidden_size, eps=1e-6)
-
-        self.attn = Attention(dim=config.hidden_size, num_heads=config.num_heads)
-        self.mlp = MLP(dim=config.hidden_size, hidden_dim=config.intermediate_size)
-
-    def __call__(self, hidden_states, cu_seqlens, rotary_pos_emb) -> mx.array:
-        hidden_states = hidden_states + self.attn(
-            self.norm1(hidden_states),
-            cu_seqlens=cu_seqlens,
-            rotary_pos_emb=rotary_pos_emb,
-        )
-        hidden_states = hidden_states + self.mlp(self.norm2(hidden_states))
-        return hidden_states
-
-
-class VisionModel(nn.Module):
-
-    def __init__(self, config: VisionConfig) -> None:
-        super().__init__()
-        self.config = config
-        self.model_type = config.model_type
-        if self.model_type != "qwen2_5_vl":
-            raise ValueError(f"Unsupported model type: {self.model_type}")
-        self.spatial_merge_size = config.spatial_merge_size
-
-        self.patch_embed = PatchEmbed(
-            patch_size=config.patch_size,
-            temporal_patch_size=config.temporal_patch_size,
-            in_channels=config.in_channels,
-            hidden_size=config.hidden_size,
-        )
-
-        self.window_size = config.window_size
-        self.patch_size = config.patch_size
-        self.spatial_merge_unit = self.spatial_merge_size * self.spatial_merge_size
-        self.fullatt_block_indexes = config.fullatt_block_indexes
-        head_dim = config.hidden_size // config.num_heads
-        self.rotary_pos_emb = VisionRotaryEmbedding(head_dim // 2)
-
-        self.blocks = [Qwen2VLVisionBlock(config) for _ in range(config.depth)]
-        self.merger = PatchMerger(
-            dim=config.out_hidden_size, context_dim=config.hidden_size
-        )
-
-    def rot_pos_emb(self, grid_thw):
-        pos_ids = []
-
-        for t, h, w in grid_thw.tolist():
-            hpos_ids = mx.expand_dims(mx.arange(h), 1)
-            hpos_ids = mx.repeat(hpos_ids, w, axis=1)
-            hpos_ids = hpos_ids.reshape(
-                h // self.spatial_merge_size,
-                self.spatial_merge_size,
-                w // self.spatial_merge_size,
-                self.spatial_merge_size,
-            )
-            hpos_ids = mx.transpose(hpos_ids, (0, 2, 1, 3))
-            hpos_ids = hpos_ids.flatten()
-
-            wpos_ids = mx.expand_dims(mx.arange(w), 0)
-            wpos_ids = mx.repeat(wpos_ids, h, axis=0)
-            wpos_ids = wpos_ids.reshape(
-                h // self.spatial_merge_size,
-                self.spatial_merge_size,
-                w // self.spatial_merge_size,
-                self.spatial_merge_size,
-            )
-            wpos_ids = mx.transpose(wpos_ids, (0, 2, 1, 3))
-            wpos_ids = wpos_ids.flatten()
-
-            stacked_pos_ids = mx.stack([hpos_ids, wpos_ids], axis=-1)
-            pos_ids.append(mx.tile(stacked_pos_ids, (t, 1)))
-
-        pos_ids = mx.concatenate(pos_ids, axis=0)
-        max_grid_size = mx.max(grid_thw[:, 1:])
-        rotary_pos_emb_full = self.rotary_pos_emb(max_grid_size)
-        rotary_pos_emb = rotary_pos_emb_full[pos_ids]
-
-        return rotary_pos_emb.reshape(pos_ids.shape[0], -1)
-
-    def get_window_index(self, grid_thw):
-        window_index = []
-        cu_window_seqlens = [0]
-        window_index_id = 0
-        vit_merger_window_size = (
-            self.window_size // self.spatial_merge_size // self.patch_size
-        )
-
-        for grid_t, grid_h, grid_w in grid_thw.tolist():
-            llm_grid_h = grid_h // self.spatial_merge_size
-            llm_grid_w = grid_w // self.spatial_merge_size
-
-            index = mx.arange(grid_t * llm_grid_h * llm_grid_w).reshape(
-                grid_t, llm_grid_h, llm_grid_w
-            )
-
-            pad_h = vit_merger_window_size - llm_grid_h % vit_merger_window_size
-            pad_w = vit_merger_window_size - llm_grid_w % vit_merger_window_size
-            num_windows_h = (llm_grid_h + pad_h) // vit_merger_window_size
-            num_windows_w = (llm_grid_w + pad_w) // vit_merger_window_size
-
-            # Replace F.pad with np.pad
-            index_padded = mx.pad(
-                index,
-                ((0, 0), (0, pad_h), (0, pad_w)),
-                mode="constant",
-                constant_values=-100,
-            )
-
-            index_padded = index_padded.reshape(
-                grid_t,
-                num_windows_h,
-                vit_merger_window_size,
-                num_windows_w,
-                vit_merger_window_size,
-            )
-
-            # Replace permute with np.transpose
-            index_padded = mx.transpose(index_padded, (0, 1, 3, 2, 4)).reshape(
-                grid_t,
-                num_windows_h * num_windows_w,
-                vit_merger_window_size,
-                vit_merger_window_size,
-            )
-
-            # Replace torch operations with numpy
-            seqlens = mx.sum(index_padded != -100, axis=(2, 3)).reshape(-1)
-            index_padded = index_padded.reshape(-1)
-            index = np.where(index_padded != -100)[
-                0
-            ].tolist()  # [i for i, x in enumerate(index_padded) if x != -100]
-            index_new = index_padded[index]
-
-            window_index.append(index_new + window_index_id)
-            cu_seqlens_tmp = (
-                mx.cumsum(seqlens, axis=0) * self.spatial_merge_unit
-                + cu_window_seqlens[-1]
-            )
-            cu_window_seqlens.extend(cu_seqlens_tmp.tolist())
-            window_index_id += int(grid_t * llm_grid_h * llm_grid_w)
-
-        # Replace torch.cat with np.concatenate
-        window_index = mx.concatenate(window_index, axis=0)
-        cu_window_seqlens = mx.array(cu_window_seqlens)
-
-        return window_index, cu_window_seqlens
-
-    def __call__(
-        self,
-        hidden_states: mx.array,
-        grid_thw: mx.array,
-        output_hidden_states: Optional[bool] = None,
-    ) -> mx.array:
-
-        hidden_states = self.patch_embed(hidden_states)
-        rotary_pos_emb = self.rot_pos_emb(grid_thw)
-        window_index, cu_window_seqlens = self.get_window_index(grid_thw)
-
-        # Get indices of first occurrence of each unique value
-        seen = set()
-        idx = []
-        for i, x in enumerate(cu_window_seqlens):
-            if x not in seen:
-                seen.add(x)
-                idx.append(i)
-
-        idx = mx.array(idx, dtype=mx.int32)
-        cu_window_seqlens = cu_window_seqlens[idx]
-
-        seq_len, _ = hidden_states.shape
-        hidden_states = hidden_states.reshape(
-            seq_len // self.spatial_merge_unit, self.spatial_merge_unit, -1
-        )
-        hidden_states = hidden_states[window_index, :, :]
-        hidden_states = hidden_states.reshape(seq_len, -1)
-        rotary_pos_emb = rotary_pos_emb.reshape(
-            seq_len // self.spatial_merge_unit, self.spatial_merge_unit, -1
-        )
-        rotary_pos_emb = rotary_pos_emb[window_index, :, :]
-        rotary_pos_emb = rotary_pos_emb.reshape(seq_len, -1)
-
-        # Assuming grid_thw has shape (batch_size, 3)
-        batch_size = grid_thw.shape[0]
-
-        # Calculate cu_seqlens for each item in the batch
-        cu_seqlens = []
-        for i in range(batch_size):
-            seq_len = grid_thw[i, 1] * grid_thw[i, 2]
-            cu_seqlens.append(mx.repeat(seq_len, grid_thw[i, 0]))
-
-        # Concatenate the cu_seqlens for all items in the batch
-        cu_seqlens = mx.concatenate(cu_seqlens)
-
-        cu_seqlens = mx.cumsum(cu_seqlens.astype(mx.int32), axis=0)
-        cu_seqlens = mx.pad(cu_seqlens, (1, 0), mode="constant", constant_values=0)
-
-        encoder_states = (hidden_states,) if output_hidden_states else None
-
-        for layer_num, blk in enumerate(self.blocks):
-            if layer_num in self.fullatt_block_indexes:
-                cu_seqlens_now = cu_seqlens
-            else:
-                cu_seqlens_now = cu_window_seqlens
-
-            hidden_states = blk(
-                hidden_states, cu_seqlens=cu_seqlens_now, rotary_pos_emb=rotary_pos_emb
-            )
-
-            if output_hidden_states:
-                encoder_states = encoder_states + (hidden_states,)
-
-        hidden_states = self.merger(hidden_states)
-        reverse_indices = mx.argsort(window_index, axis=0)
-        hidden_states = hidden_states[reverse_indices, :]
-        return hidden_states
-
-    def sanitize(self, weights):
-        sanitized_weights = {}
-        for k, v in weights.items():
-            if "position_ids" in k:
-                # Remove unused position_ids
-                continue
-            elif "patch_embed.proj.weight" in k:
-                # PyTorch conv2d weight tensors have shape:
-                #   [out_channels, in_channels, kH, KW]
-                # MLX conv2d expects the weight be of shape:
-                #   [out_channels, kH, KW, in_channels]
-                if check_array_shape(v):
-                    sanitized_weights[k] = v
-                else:
-                    sanitized_weights[k] = v.transpose(0, 2, 3, 4, 1)
-            else:
-                sanitized_weights[k] = v
-
-        return sanitized_weights

nexaai/binds/nexa_mlx/py-lib/vlm/modeling/models/qwen2_vl/__init__.py

@@ -1,2 +0,0 @@
-from .config import ModelConfig, TextConfig, VisionConfig
-from .qwen2_vl import LanguageModel, Model, VisionModel

nexaai/binds/nexa_mlx/py-lib/vlm/modeling/models/qwen2_vl/config.py

@@ -1,104 +0,0 @@
-import inspect
-from dataclasses import dataclass
-from typing import Dict, List, Optional, Union
-
-
-@dataclass
-class VisionConfig:
-    model_type: str = "qwen2_vl"
-    depth: int = 32
-    embed_dim: int = 1280
-    hidden_size: int = 1536
-    num_heads: int = 16
-    image_size: int = 384
-    patch_size: int = 14
-    vocab_size: int = 32000
-    mlp_ratio: float = 4.0
-    in_channels: int = 3
-    layer_norm_eps: float = 1e-6
-    spatial_patch_size: int = 14
-    spatial_merge_size: int = 2
-    temporal_patch_size: int = 2
-
-    @classmethod
-    def from_dict(cls, params):
-        return cls(
-            **{
-                k: v
-                for k, v in params.items()
-                if k in inspect.signature(cls).parameters
-            }
-        )
-
-
-@dataclass
-class TextConfig:
-    model_type: str
-    hidden_size: int
-    num_hidden_layers: int
-    intermediate_size: int
-    num_attention_heads: int
-    rms_norm_eps: float
-    vocab_size: int
-    num_key_value_heads: Optional[int] = 8
-    max_position_embeddings: Optional[int] = 40960
-    rope_theta: float = 1000000.0
-    rope_traditional: bool = False
-    rope_scaling: Optional[Dict[str, Union[float, str]]] = None
-    tie_word_embeddings: bool = False
-    sliding_window: int = 32768
-    use_sliding_window: bool = False
-    use_cache: bool = True
-
-    def __post_init__(self):
-        if self.num_key_value_heads is None:
-            self.num_key_value_heads = self.num_attention_heads
-
-        if self.rope_scaling:
-            required_keys = {"mrope_section", "type"}
-            if not all(key in self.rope_scaling for key in required_keys):
-                raise ValueError(f"rope_scaling must contain keys {required_keys}")
-
-            if not self.rope_scaling["type"] in ["mrope", "default"]:
-                raise ValueError(f"rope_scaling type must be 'mrope' or 'default'")
-
-    @classmethod
-    def from_dict(cls, params):
-        return cls(
-            **{
-                k: v
-                for k, v in params.items()
-                if k in inspect.signature(cls).parameters
-            }
-        )
-
-
-@dataclass
-class ModelConfig:
-    text_config: TextConfig
-    vision_config: VisionConfig
-    model_type: str
-    ignore_index: int = -100
-    image_token_id: int = 151655
-    video_token_id: int = 151656
-    vision_start_token_id: int = 151652
-    vision_feature_select_strategy: str = "default"
-    vision_feature_layer: int = -2
-    vocab_size: int = 32000
-    eos_token_id: Optional[List[int]] = None
-
-    @classmethod
-    def from_dict(cls, params):
-        # Copy text config parameters from root level
-        excluded_keys = {"vision_config"}
-        params["text_config"] = dict(
-            filter(lambda x: x[0] not in excluded_keys, params.items())
-        )
-
-        return cls(
-            **{
-                k: v
-                for k, v in params.items()
-                if k in inspect.signature(cls).parameters
-            }
-        )