Spaces:
Sleeping
Sleeping
kz209
commited on
Commit
·
8e22bd4
1
Parent(s):
331a464
update
Browse files- utils/model.py +1 -1
utils/model.py
CHANGED
@@ -56,7 +56,7 @@ class Model(torch.nn.Module):
|
|
56 |
sampling_params = SamplingParams(
|
57 |
temperature=temp,
|
58 |
max_tokens=max_length,
|
59 |
-
top_p=0.95 if do_sample else 1.0,
|
60 |
stop_token_ids=[self.tokenizer.eos_token_id]
|
61 |
)
|
62 |
outputs = self.llm.generate(content_list, sampling_params)
|
|
|
56 |
sampling_params = SamplingParams(
|
57 |
temperature=temp,
|
58 |
max_tokens=max_length,
|
59 |
+
#top_p=0.95 if do_sample else 1.0,
|
60 |
stop_token_ids=[self.tokenizer.eos_token_id]
|
61 |
)
|
62 |
outputs = self.llm.generate(content_list, sampling_params)
|