YC-Chen commited on
Commit
09c1838
1 Parent(s): 28e4465

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -91,13 +91,13 @@ llm = LLM(
91
  gpu_memory_utilization=0.7
92
  )
93
 
94
- instance_end_token_id = llm.get_tokenizer().convert_token_to_ids('<|im_end|>')
95
  params = SamplingParams(
96
- temperature=0.01,
97
- top_p=0.01,
98
  max_tokens=4096,
99
  repetition_penalty=1.1,
100
- stop_token_ids=[instance_end_token_id]
101
  )
102
 
103
  def _inference(prompt, llm, params):
 
91
  gpu_memory_utilization=0.7
92
  )
93
 
94
+ turn_end_token_id = 61876 # <|im_end|>
95
  params = SamplingParams(
96
+ temperature=0.001,
97
+ top_p=0.001,
98
  max_tokens=4096,
99
  repetition_penalty=1.1,
100
+ stop_token_ids=[turn_end_token_id]
101
  )
102
 
103
  def _inference(prompt, llm, params):