Image-Text-to-Text
Transformers
Safetensors
English
nanbeige_vlm
vlm
nanbeige
siglip
conversational
custom_code
Nanbeige4.1-VLM-Base / configuration_nanbeige_vlm.py
SkyAsl's picture
Upload configuration_nanbeige_vlm.py with huggingface_hub
ad2b447 verified
Raw
History Blame
623 Bytes
from transformers import PretrainedConfig
class NanbeigeVLMConfig(PretrainedConfig):
"""
Configuration for NanbeigeVLM.
Args:
vision_model_id: HuggingFace model ID for the vision tower.
llm_model_id: HuggingFace model ID for the language model.
"""
model_type = "nanbeige_vlm"
def __init__(
self,
vision_model_id: str = "google/siglip-so400m-patch14-384",
llm_model_id: str = "Nanbeige/Nanbeige4.1-3B",
**kwargs,
):
super().__init__(**kwargs)
self.vision_model_id = vision_model_id
self.llm_model_id = llm_model_id