from transformers import Qwen3VLConfig, AutoConfig


class NasChildVLConfig(Qwen3VLConfig):
    model_type = "nas-child-vl"

    def __init__(self, nas_layer_config=None, **kwargs):
        kwargs.pop("model_type", None)

        _from_kwargs = kwargs.pop("nas_layer_config", None)
        if nas_layer_config is None:
            nas_layer_config = _from_kwargs

        super().__init__(**kwargs)

        if nas_layer_config is not None and len(nas_layer_config) > 0:
            self.nas_layer_config = nas_layer_config
        elif hasattr(self, "nas_layer_config") and self.nas_layer_config:
            pass
        else:
            self.nas_layer_config = []

    def to_dict(self):
        output = super().to_dict()
        output["nas_layer_config"] = self.nas_layer_config
        if "text_config" in output and isinstance(output["text_config"], dict):
            output["text_config"].pop("nas_layer_config", None)
        return output


try:
    AutoConfig.register("nas-child-vl", NasChildVLConfig)
except Exception:
    pass