LLMFactCheck

KoslickiLab · Feb 18, 2024 · 4c099ba · 4c099ba
1 parent db3627a
commit 4c099ba
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/get_result.py b/src/get_result.py
@@ -29,9 +29,11 @@ def get_result(model_info, prompt, model_type):
         result_text = ""
         for chunk in prompt_chunks:
             # 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)
             result_text += response["choices"][0]["text"]
+            print(result_text)
         return result_text
 
     else: