LuckyStarr commited on
Commit
08d05ca
1 Parent(s): e3510fd

correct a typo in the code snippet

Browse files
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)