Method match

lmdeploy/model.py:230–238

Return the model_name that was registered to MODELS. Args: model_path (str): the model path used for matching.

(cls, model_path: str, **kwargs)

    @classmethod
    def match(cls, model_path: str, **kwargs) -> str | None:
        """Return the model_name that was registered to MODELS.

        Args:
            model_path (str): the model path used for matching.
        """
        path = model_path.lower()
        if 'cogvlm' in path and 'cogvlm2' not in path:
            return 'cogvlm'


@MODELS.register_module(name='vicuna')