Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-16 16:27:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

Feat/update default revision (#1553)

Browse Source
This commit is contained in:
Gongsheng Li
2025-12-05 15:24:42 +08:00
committed by suluyan
parent f35932e90b
commit c2f91d80fe
2 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
modelscope/utils/constant.py
View File

@@ -507,7 +507,7 @@ class Frameworks(object):
REPO_TYPE_MODEL = 'model' REPO_TYPE_MODEL = 'model'
REPO_TYPE_DATASET = 'dataset' REPO_TYPE_DATASET = 'dataset'
REPO_TYPE_SUPPORT = [REPO_TYPE_MODEL, REPO_TYPE_DATASET] REPO_TYPE_SUPPORT = [REPO_TYPE_MODEL, REPO_TYPE_DATASET]
DEFAULT_MODEL_REVISION = None DEFAULT_MODEL_REVISION = 'master'
MASTER_MODEL_BRANCH = 'master' MASTER_MODEL_BRANCH = 'master'
DEFAULT_REPOSITORY_REVISION = 'master' DEFAULT_REPOSITORY_REVISION = 'master'
DEFAULT_DATASET_REVISION = 'master' DEFAULT_DATASET_REVISION = 'master'

4
tests/hub/test_hub_revision_release_mode.py
View File

@@ -108,6 +108,9 @@ class HubRevisionTest(unittest.TestCase):
cache_dir=temp_cache_dir) cache_dir=temp_cache_dir)
assert os.path.exists(file_path) assert os.path.exists(file_path)
@unittest.skip(
'snapshot_download cannot exercise release-mode auto-selection while'
' DEFAULT_MODEL_REVISION forces a concrete branch (master).')
def test_snapshot_download_revision(self): def test_snapshot_download_revision(self):
with mock.patch.dict(os.environ, self.modified_environ, clear=True): with mock.patch.dict(os.environ, self.modified_environ, clear=True):
self.prepare_repo_data_and_tag() self.prepare_repo_data_and_tag()
@@ -205,6 +208,7 @@ class HubRevisionTest(unittest.TestCase):
model_file_download( model_file_download(
self.model_id, self.model_id,
download_model_file_name2, download_model_file_name2,
revision=None,
cache_dir=temp_cache_dir) cache_dir=temp_cache_dir)
version.__release_datetime__ = t2 version.__release_datetime__ = t2
logger.info('Setting __release_datetime__ to: %s' % t2) logger.info('Setting __release_datetime__ to: %s' % t2)