mirror of
https://github.com/modelscope/modelscope.git
synced 2026-02-24 04:01:10 +01:00
fix downloading
This commit is contained in:
@@ -16,16 +16,16 @@ from modelscope.utils.repo_utils import (CommitInfo, CommitOperation,
|
||||
CommitOperationAdd)
|
||||
|
||||
ignore_file_pattern = [
|
||||
r'\w+\.bin',
|
||||
r'\w+\.safetensors',
|
||||
r'\w+\.pth',
|
||||
r'\w+\.pt',
|
||||
r'\w+\.h5',
|
||||
r'\w+\.ckpt',
|
||||
r'\w+\.zip',
|
||||
r'\w+\.onnx',
|
||||
r'\w+\.tar',
|
||||
r'\w+\.gz',
|
||||
r'*.bin',
|
||||
r'*.safetensors',
|
||||
r'*.pth',
|
||||
r'*.pt',
|
||||
r'*.h5',
|
||||
r'*.ckpt',
|
||||
r'*.zip',
|
||||
r'*.onnx',
|
||||
r'*.tar',
|
||||
r'*.gz',
|
||||
]
|
||||
|
||||
|
||||
|
||||
40
tests/hub/test_download_tokenizer.py
Normal file
40
tests/hub/test_download_tokenizer.py
Normal file
@@ -0,0 +1,40 @@
|
||||
# Copyright (c) Alibaba, Inc. and its affiliates.
|
||||
import os.path
|
||||
import shutil
|
||||
import tempfile
|
||||
import unittest
|
||||
|
||||
from modelscope import snapshot_download
|
||||
|
||||
|
||||
class TestDownloadTokenizer(unittest.TestCase):
|
||||
|
||||
def setUp(self):
|
||||
temporary_dir = tempfile.mkdtemp()
|
||||
self.work_dir = temporary_dir
|
||||
|
||||
def tearDown(self):
|
||||
shutil.rmtree(self.work_dir, ignore_errors=True)
|
||||
|
||||
def test_download_tokenizer(self):
|
||||
ignore_file_pattern = [
|
||||
r'*.bin',
|
||||
r'*.safetensors',
|
||||
r'*.pth',
|
||||
r'*.pt',
|
||||
r'*.h5',
|
||||
r'*.ckpt',
|
||||
r'*.zip',
|
||||
r'*.onnx',
|
||||
r'*.tar',
|
||||
r'*.gz',
|
||||
]
|
||||
model_dir = snapshot_download(
|
||||
'Qwen/Qwen3-0.6B', cache_dir=self.work_dir, ignore_file_pattern=ignore_file_pattern)
|
||||
self.assertTrue(model_dir is not None)
|
||||
self.assertTrue(os.path.exists(os.path.join(model_dir, 'tokenizer.json')))
|
||||
self.assertFalse(os.path.exists(os.path.join(model_dir, 'model.safetensors')))
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user