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