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

Add missing import (#497)

Browse Source 
* Add missing import

* format

---------

Co-authored-by: Awni Hannun <awni@apple.com>
This commit is contained in:
Y4hL 2024-02-27 23:27:08 +02:00 committed by GitHub
parent 95f82e67a2
commit 676e574eff
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
llms/mlx_lm/generate.py
View File

@ -115,7 +115,14 @@ def main(args):
formatter = colorprint_by_t0 if args.colorize else None formatter = colorprint_by_t0 if args.colorize else None
generate( generate(
model, tokenizer, prompt, args.temp, args.max_tokens, True, formatter=formatter, top_p=args.top_p model,
tokenizer,
prompt,
args.temp,
args.max_tokens,
True,
formatter=formatter,
top_p=args.top_p,
) )

1
llms/mlx_lm/merge.py
View File

@ -3,6 +3,7 @@
import argparse import argparse
import glob import glob
import json import json
import shutil
from pathlib import Path from pathlib import Path
import mlx.core as mx import mlx.core as mx