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

prompt parameter (#291)

Browse Source
This commit is contained in:
Anjor Kanekar 2024-01-11 14:04:57 +00:00 committed by GitHub
parent 7380ebfb0d
commit e74889d0fa
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lora/lora.py
View File

@ -263,9 +263,9 @@ def train(model, train_set, val_set, optimizer, loss, tokenizer, args):
def generate(model, prompt, tokenizer, args): def generate(model, prompt, tokenizer, args):
print(args.prompt, end="", flush=True) print(prompt, end="", flush=True)
prompt = mx.array(tokenizer.encode(args.prompt)) prompt = mx.array(tokenizer.encode(prompt))
tokens = [] tokens = []
skip = 0 skip = 0