mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-29 11:02:21 +00:00
fixes
This commit is contained in:
parent
a5d6ab16b2
commit
9b58624479
2 changed files with 11 additions and 4 deletions
|
|
@ -253,7 +253,8 @@ class MetaReferenceInferenceImpl(
|
|||
def impl():
|
||||
stop_reason = None
|
||||
|
||||
for token_result in self.generator.completion(request):
|
||||
for token_results in self.generator.completion([request]):
|
||||
token_result = token_results[0]
|
||||
if token_result.token == tokenizer.eot_id:
|
||||
stop_reason = StopReason.end_of_turn
|
||||
text = ""
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue