Update README.md
Browse files
README.md
CHANGED
|
@@ -37,7 +37,7 @@ def generate_response (text_input="Biological materials offer amazing",
|
|
| 37 |
top_k = top_k,
|
| 38 |
top_p =top_p,
|
| 39 |
num_return_sequences = num_return_sequences, eos_token_id=eos_token_id,
|
| 40 |
-
do_sample =True
|
| 41 |
repetition_penalty=repetition_penalty,
|
| 42 |
)
|
| 43 |
return tokenizer.batch_decode(outputs[:,inputs.shape[1]:].detach().cpu().numpy(), skip_special_tokens=True)
|
|
|
|
| 37 |
top_k = top_k,
|
| 38 |
top_p =top_p,
|
| 39 |
num_return_sequences = num_return_sequences, eos_token_id=eos_token_id,
|
| 40 |
+
do_sample =True,
|
| 41 |
repetition_penalty=repetition_penalty,
|
| 42 |
)
|
| 43 |
return tokenizer.batch_decode(outputs[:,inputs.shape[1]:].detach().cpu().numpy(), skip_special_tokens=True)
|