Skip to content

Commit

Permalink
LLMFactCheck
Browse files Browse the repository at this point in the history
  • Loading branch information
mlupei committed Feb 18, 2024
1 parent d5198f0 commit d8948db
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 @@ -31,7 +31,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.5,
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
Expand Down

0 comments on commit d8948db

Please sign in to comment.