heanu LuckyStarr commited on
Commit
9448ccd
1 Parent(s): e3510fd

correct a typo in the code snippet (#7)

Browse files

- correct a typo in the code snippet (08d05cab5a671e2b96d1da8142d386b1379d237f)


Co-authored-by: Lucky Starr <LuckyStarr@users.noreply.huggingface.co>

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -59,7 +59,7 @@ def generate(situation_narrative, role_instruction, conversation_history):
59
  conversation_history: the previous utterances in the conversation in a list
60
  """
61
 
62
- input_text = set_input(role_narrative, role_instruction, conversation_history)
63
 
64
  inputs = tokenizer([input_text], return_tensors="pt").to(device)
65
  outputs = model.generate(inputs["input_ids"], max_new_tokens=128, temperature=1.0, top_p=.95, do_sample=True)
 
59
  conversation_history: the previous utterances in the conversation in a list
60
  """
61
 
62
+ input_text = set_input(situation_narrative, role_instruction, conversation_history)
63
 
64
  inputs = tokenizer([input_text], return_tensors="pt").to(device)
65
  outputs = model.generate(inputs["input_ids"], max_new_tokens=128, temperature=1.0, top_p=.95, do_sample=True)