Class _MsKernelApi

modelscope/utils/hf_util/patcher.py:555–630

Minimal `HfApi` look-alike that forwards to ModelScope. Only the handful of methods that `kernels` actually calls are implemented.

class _MsKernelApi:
    """Minimal `HfApi` look-alike that forwards to ModelScope. Only the
    handful of methods that `kernels` actually calls are implemented.
    """

    def snapshot_download(self,
                          repo_id,
                          *,
                          allow_patterns=None,
                          ignore_patterns=None,
                          cache_dir=None,
                          revision=None,
                          local_files_only=False,
                          **kwargs):
        from modelscope import snapshot_download as ms_snapshot_download
        if kwargs.get('repo_type') == 'kernel':
            kwargs['repo_type'] = 'model'
        return ms_snapshot_download(
            repo_id,
            revision=_ms_revision(revision),
            cache_dir=cache_dir,
            local_files_only=local_files_only,
            allow_patterns=allow_patterns,
            ignore_patterns=ignore_patterns,
            **kwargs)

    def list_repo_tree(self,
                       repo_id,
                       *,
                       path_in_repo=None,
                       revision=None,
                       **kwargs):
        from huggingface_hub.hf_api import RepoFolder
        from modelscope.hub.api import HubApi
        entries = HubApi().get_model_files(
            repo_id,
            revision=_ms_revision(revision),
            root=path_in_repo,
            recursive=False)
        folders = []
        for entry in entries:
            if entry.get('Type') != 'tree':
                continue
            path = entry.get('Path') or entry.get('Name')
            folders.append(RepoFolder(path=path, oid='', last_commit=None))
        return folders

    def file_exists(self, repo_id, filename, *, revision=None, **kwargs):
        from modelscope.hub.api import HubApi
        return HubApi().file_exists(
            repo_id, filename, revision=_ms_revision(revision))

    def list_repo_refs(self, repo_id, **kwargs):
        from huggingface_hub.hf_api import GitRefInfo
        from modelscope.hub.api import HubApi
        branches, tags = HubApi().get_model_branches_and_tags(repo_id)
        # `target_commit` doubles as the revision in later calls, so reuse
        # the branch/tag name (ModelScope accepts it as a revision).