Update README.md
Browse files
README.md
CHANGED
|
@@ -58,7 +58,6 @@ prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_genera
|
|
| 58 |
gen_config = {
|
| 59 |
"max_new_tokens": 1024,
|
| 60 |
"do_sample": False,
|
| 61 |
-
"stop_strings": ["```output"], # Generate until Python code block is complete
|
| 62 |
"tokenizer": pipe.tokenizer,
|
| 63 |
}
|
| 64 |
|
|
|
|
| 58 |
gen_config = {
|
| 59 |
"max_new_tokens": 1024,
|
| 60 |
"do_sample": False,
|
|
|
|
| 61 |
"tokenizer": pipe.tokenizer,
|
| 62 |
}
|
| 63 |
|