hannayukhymenko commited on
Commit
2d4fd7a
·
verified ·
1 Parent(s): 3aae635

fix eos token id

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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,107],
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, 107],
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",