Unverified Commit 53d7c392 authored by Aviad Rossmann's avatar Aviad Rossmann Committed by GitHub
Browse files

Update sampling_metadata.py (#21937)


Signed-off-by: default avatarAviad Rossmann <aviadr@neureality.ai>
parent 61dcc280
......@@ -539,37 +539,37 @@ class SamplingTensors:
temperatures_t = torch.tensor(
temperatures,
device="cpu",
dtype=dtype,
dtype=torch.float32,
pin_memory=pin_memory,
)
top_ps_t = torch.tensor(
top_ps,
device="cpu",
dtype=dtype,
dtype=torch.float32,
pin_memory=pin_memory,
)
min_ps_t = torch.tensor(
min_ps,
device="cpu",
dtype=dtype,
dtype=torch.float32,
pin_memory=pin_memory,
)
presence_penalties_t = torch.tensor(
presence_penalties,
device="cpu",
dtype=dtype,
dtype=torch.float32,
pin_memory=pin_memory,
)
frequency_penalties_t = torch.tensor(
frequency_penalties,
device="cpu",
dtype=dtype,
dtype=torch.float32,
pin_memory=pin_memory,
)
repetition_penalties_t = torch.tensor(
repetition_penalties,
device="cpu",
dtype=dtype,
dtype=torch.float32,
pin_memory=pin_memory,
)
top_ks_t = torch.tensor(
......
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