yinghy18 commited on
Commit
84a32a0
1 Parent(s): 1795aa2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -33,7 +33,7 @@ GENIE is co-developed by the groups of Sheng Yu (https://www.stat.tsinghua.edu.c
33
  ```python
34
  from vllm import LLM, SamplingParams
35
 
36
- PROMPT_TEMPLATE = "Human:\n{query}\n\n Assistant:"
37
  sampling_params = SamplingParams(temperature=temperature, max_tokens=max_new_token)
38
  EHR = ['xxxxx1','xxxxx2']
39
  texts = [PROMPT_TEMPLATE.format(query=k) for k in EHR]
 
33
  ```python
34
  from vllm import LLM, SamplingParams
35
 
36
+ PROMPT_TEMPLATE = "Human:\n{query}\n\n Assistant:\n" #please stricly follow for now
37
  sampling_params = SamplingParams(temperature=temperature, max_tokens=max_new_token)
38
  EHR = ['xxxxx1','xxxxx2']
39
  texts = [PROMPT_TEMPLATE.format(query=k) for k in EHR]