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

ok Merge branch 'main' of github.com:ml-explore/mlx-examples into add_modelscope

Browse Source
This commit is contained in:
xingjun.wang 2025-01-08 02:46:56 +08:00
commit 6bb7163013
2 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
llms/mlx_lm/generate.py
View File

@ -45,7 +45,7 @@ def setup_arg_parser():
parser.add_argument( parser.add_argument(
"--extra-eos-token", "--extra-eos-token",
type=str, type=str,
default=None, default=(),
nargs="+", nargs="+",
help="Add tokens in the list of eos tokens that stop generation.", help="Add tokens in the list of eos tokens that stop generation.",
) )

4
llms/mlx_lm/utils.py
View File

@ -697,12 +697,10 @@ def upload_to_hub(path: str, upload_repo: str, hf_path: str):
api = HfApi() api = HfApi()
api.create_repo(repo_id=upload_repo, exist_ok=True) api.create_repo(repo_id=upload_repo, exist_ok=True)
api.upload_folder( api.upload_large_folder(
folder_path=path, folder_path=path,
repo_id=upload_repo, repo_id=upload_repo,
repo_type="model", repo_type="model",
multi_commits=True,
multi_commits_verbose=True,
) )
print(f"Upload successful, go to https://huggingface.co/{upload_repo} for details.") print(f"Upload successful, go to https://huggingface.co/{upload_repo} for details.")