knowledge_llama.py 392 B

1234567891011121314
  1. from evaluator import *
  2. DESCRIPTION = "Test the knowledge cutoff of the model to see if it knows the LLAMA-2 hidden dimension size."
  3. TAGS = ['explain']
  4. question = """
  5. What is the hidden dimension of LLAMA-2 70b?
  6. """
  7. TestLlamaKnowledge = question >> LLMRun() >> (SubstringEvaluator("8192") | SubstringEvaluator("8,192"))
  8. if __name__ == "__main__":
  9. print(run_test(TestLlamaKnowledge))