Commit
•
08d05ca
1
Parent(s):
e3510fd
correct a typo in the code snippet
Browse files
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(
|
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)
|