kz209 commited on
Commit
18784a2
·
1 Parent(s): d90d6fa
Files changed (1) hide show
  1. utils/model.py +1 -1
utils/model.py CHANGED
@@ -20,7 +20,7 @@ class Model():
20
  sequences = self.pipeline(
21
  "Girafatron is obsessed with giraffes, the most glorious animal on the face of this Earth. Giraftron believes all other animals are irrelevant when compared to the glorious majesty of the giraffe.\nDaniel: Hello, Girafatron!\nGirafatron:",
22
  max_length=max_length,
23
- do_sample=True,
24
  temperature=temp,
25
  num_return_sequences=1,
26
  eos_token_id=self.tokenizer.eos_token_id,
 
20
  sequences = self.pipeline(
21
  "Girafatron is obsessed with giraffes, the most glorious animal on the face of this Earth. Giraftron believes all other animals are irrelevant when compared to the glorious majesty of the giraffe.\nDaniel: Hello, Girafatron!\nGirafatron:",
22
  max_length=max_length,
23
+ do_sample=False,
24
  temperature=temp,
25
  num_return_sequences=1,
26
  eos_token_id=self.tokenizer.eos_token_id,