Skip to content

Commit

Permalink
LLMFactCheck
Browse files Browse the repository at this point in the history
  • Loading branch information
mlupei committed Feb 28, 2024
1 parent 9c19f65 commit 30104a3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/get_result.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,7 +38,7 @@ def get_result(model_info, prompt, model_type):
# Interact with the Llama model
print(chunk)
response = model(prompt=chunk, max_tokens=256, temperature=0.01,
top_p=0.95, repeat_penalty=1.2, top_k=150, echo=True)
top_p=0.95, repeat_penalty=1.2, top_k=150, echo=False)
result_text += response["choices"][0]["text"]
print(result_text)
return result_text, prompt_template
Expand Down

0 comments on commit 30104a3

Please sign in to comment.