fix eos token id
Browse files
README.md
CHANGED
|
@@ -87,7 +87,7 @@ generation_params = GenerationConfig(
|
|
| 87 |
top_k=25,
|
| 88 |
top_p=1,
|
| 89 |
repetition_penalty=1.1,
|
| 90 |
-
eos_token_id=[1,
|
| 91 |
do_sample=True
|
| 92 |
)
|
| 93 |
```
|
|
@@ -147,7 +147,7 @@ sampling_params = SamplingParams(
|
|
| 147 |
top_k=25,
|
| 148 |
top_p=1,
|
| 149 |
repetition_penalty=1.1,
|
| 150 |
-
stop_token_ids=[1,
|
| 151 |
)
|
| 152 |
llm = LLM(
|
| 153 |
model="INSAIT-Institute/MamayLM-Gemma-3-12B-IT-v1.0",
|
|
|
|
| 87 |
top_k=25,
|
| 88 |
top_p=1,
|
| 89 |
repetition_penalty=1.1,
|
| 90 |
+
# eos_token_id=[1,106],
|
| 91 |
do_sample=True
|
| 92 |
)
|
| 93 |
```
|
|
|
|
| 147 |
top_k=25,
|
| 148 |
top_p=1,
|
| 149 |
repetition_penalty=1.1,
|
| 150 |
+
stop_token_ids=[1, 106],
|
| 151 |
)
|
| 152 |
llm = LLM(
|
| 153 |
model="INSAIT-Institute/MamayLM-Gemma-3-12B-IT-v1.0",
|