Add files via upload
This commit is contained in:
committed by
GitHub
parent
ffa49a7a1c
commit
dd1186c029
@@ -57,7 +57,7 @@ def logits_to_probs(
|
||||
logits = logits / max(temperature, 1e-5)
|
||||
|
||||
if top_k is not None:
|
||||
v, _ = torch.topk(logits, min(top_k, logits.size(-1)))
|
||||
v, _ = torch.topk(logits, top_k)
|
||||
pivot = v.select(-1, -1).unsqueeze(-1)
|
||||
logits = torch.where(logits < pivot, inf_tensor_value, logits)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user