zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-29 18:26:37 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
xingjun.wang 2025-01-06 00:25:52 +08:00
parent 9d86b5b744
commit 5c535a28c0
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
llms/mlx_lm/utils.py
View File

@ -17,6 +17,7 @@ from typing import Any, Callable, Dict, Generator, List, Optional, Tuple, Type,
import mlx.core as mx import mlx.core as mx
import mlx.nn as nn import mlx.nn as nn
if os.getenv('MLX_USE_MODELSCOPE', 'False').lower() == 'true': if os.getenv('MLX_USE_MODELSCOPE', 'False').lower() == 'true':
print(">> Using ModelScope")
from modelscope import snapshot_download from modelscope import snapshot_download
else: else:
from huggingface_hub import snapshot_download from huggingface_hub import snapshot_download
@ -157,6 +158,10 @@ def get_model_path(path_or_hf_repo: str, revision: Optional[str] = None) -> Path
Path: The path to the model. Path: The path to the model.
""" """
model_path = Path(path_or_hf_repo) model_path = Path(path_or_hf_repo)
print(f'>>model_path: {model_path}')
revision = revision or 'master'
print(f'>>revision: {revision}')
if not model_path.exists(): if not model_path.exists():
try: try:
model_path = Path( model_path = Path(