Update README.md
This commit is contained in:
parent
a3cd77bcd2
commit
cac7ab316d
@ -109,6 +109,8 @@ completion_request = ChatCompletionRequest(
|
|||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
tokens = tokenizer.encode_chat_completion(completion_request).tokens
|
||||||
|
|
||||||
out_tokens, _ = generate([tokens], model, max_tokens=64, temperature=0.0, eos_id=tokenizer.instruct_tokenizer.tokenizer.eos_id)
|
out_tokens, _ = generate([tokens], model, max_tokens=64, temperature=0.0, eos_id=tokenizer.instruct_tokenizer.tokenizer.eos_id)
|
||||||
result = tokenizer.instruct_tokenizer.tokenizer.decode(out_tokens[0])
|
result = tokenizer.instruct_tokenizer.tokenizer.decode(out_tokens[0])
|
||||||
|
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user