zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-01 21:01:32 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix reference to changed option name

Browse Source
This commit is contained in:
Chime Ogbuji
2024-09-03 15:18:13 -04:00
parent 62655dfa2f
commit aa14faf22b
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
llms/mlx_lm/generate.py
View File

@@ -213,8 +213,8 @@ def main():
else:
prompt = args.prompt
if args.colorize and args.prompt_only:
raise ValueError("Cannot use --colorize with --prompt-only")
if args.colorize and not args.verbose:
raise ValueError("Cannot use --colorize when --verbose set to False")
formatter = colorprint_by_t0 if args.colorize else None
# Determine the max kv size from the kv cache or passed arguments
@@ -235,7 +235,7 @@ def main():
max_kv_size=max_kv_size,
cache_history=cache_history,
)
if args.prompt_only:
if not args.verbose:
print(response)