
from .configuration_internvl_chat import InternVLChatConfig
from transformers.utils import logging

logger = logging.get_logger(__name__)

class TinyVLAConfig(InternVLChatConfig):
    model_type = "tinyvla"
    keys_to_ignore_at_inference = ["past_key_values"]

    def __init__(
        self,
        policy_head_type='unet_diffusion_policy',
        **kwargs,
    ):
        super().__init__(**kwargs)
        self.policy_head_type = policy_head_type


from transformers import AutoConfig
AutoConfig.register("tinyvla", TinyVLAConfig)
