Update README.md
Browse files
README.md
CHANGED
|
@@ -31,13 +31,9 @@ messages = [
|
|
| 31 |
]
|
| 32 |
|
| 33 |
generate_kwargs = {
|
| 34 |
-
|
| 35 |
"do_sample": True,
|
| 36 |
-
|
| 37 |
"temperature": 0.7,
|
| 38 |
-
|
| 39 |
"max_new_tokens": 150,
|
| 40 |
-
|
| 41 |
}
|
| 42 |
|
| 43 |
pipe = pipeline("text-generation", model="bertin-project/Gromenauer-7B-Instruct", generate_kwargs=generate_kwargs)
|
|
|
|
| 31 |
]
|
| 32 |
|
| 33 |
generate_kwargs = {
|
|
|
|
| 34 |
"do_sample": True,
|
|
|
|
| 35 |
"temperature": 0.7,
|
|
|
|
| 36 |
"max_new_tokens": 150,
|
|
|
|
| 37 |
}
|
| 38 |
|
| 39 |
pipe = pipeline("text-generation", model="bertin-project/Gromenauer-7B-Instruct", generate_kwargs=generate_kwargs)
|