Update README.md
Browse files
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=
|
72 |
-
top_k=
|
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)
|