Mr-Vicky-01 commited on
Commit
2a3a26f
·
verified ·
1 Parent(s): 1d28963

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -68,10 +68,10 @@ while True:
68
  pad_token_id=151644,
69
  eos_token_id=151645,
70
  num_return_sequences=1,
71
- do_sample=True,
72
- top_k=50,
73
- temperature=0.2,
74
- top_p=0.90
75
  )
76
 
77
  generated_ids = torch.cat([generated_ids, next_token[:, -1:]], dim=1)
 
68
  pad_token_id=151644,
69
  eos_token_id=151645,
70
  num_return_sequences=1,
71
+ do_sample=False,
72
+ # top_k=5,
73
+ # temperature=0.2,
74
+ # top_p=0.90
75
  )
76
 
77
  generated_ids = torch.cat([generated_ids, next_token[:, -1:]], dim=1)