"docker/Dockerfile" did not exist on "14f301b541ffe95187ff90938c7f94f642bc5bfa"
Unverified Commit 96efb914 authored by Woosuk Kwon's avatar Woosuk Kwon Committed by GitHub
Browse files

[Model Runner V2] Fix processed logits in sample() (#37144)


Signed-off-by: default avatarWoosuk Kwon <woosuk@inferact.ai>
parent 2754231b
......@@ -178,4 +178,4 @@ class Sampler:
pos,
apply_temperature=False,
)
return sampled, logits
return sampled, processed_logits
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment