diff --git a/modelscope/hub/api.py b/modelscope/hub/api.py index 984f1454..cd3b34fc 100644 --- a/modelscope/hub/api.py +++ b/modelscope/hub/api.py @@ -130,8 +130,7 @@ class HubApi: self.upload_checker = UploadingCheck() - @staticmethod - def _get_cookies(access_token: str): + def _get_cookies(self, access_token: str): """ Get jar cookies for authentication from access_token. @@ -142,10 +141,14 @@ class HubApi: jar (CookieJar): cookies for authentication. """ from requests.cookies import RequestsCookieJar + from urllib.parse import urlparse + + domain: str = urlparse(self.endpoint).netloc if self.endpoint else get_domain() + jar = RequestsCookieJar() jar.set('m_session_id', access_token, - domain=get_domain(), + domain=domain, path='/') return jar