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

Fix for stream_generate() takes 3 positional arguments but 4 were given

Browse Source
This commit is contained in:
Ivan Fioravanti 2024-12-11 06:17:33 +01:00
parent 135c5818c1
commit 1fd2bfa5f5
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
llms/mlx_lm/chat.py
View File

@ -76,9 +76,9 @@ def main():
messages, tokenize=False, add_generation_prompt=True messages, tokenize=False, add_generation_prompt=True
) )
for response in stream_generate( for response in stream_generate(
model, model=model,
tokenizer, tokenizer=tokenizer,
prompt, prompt=prompt,
max_tokens=args.max_tokens, max_tokens=args.max_tokens,
sampler=make_sampler(args.temp, args.top_p), sampler=make_sampler(args.temp, args.top_p),
prompt_cache=prompt_cache, prompt_cache=prompt_cache,