what_is_automodel.py 446 B

1234567891011121314
  1. from evaluator import *
  2. DESCRIPTION = "Test if the model can interpret vague questions and will respond with the answer I want, not the answer that's easy to find."
  3. TAGS = ['explain']
  4. question = """
  5. What do I do to fix AutoModel.from_pretrained to make it auto model with lm head
  6. """
  7. TestWhatIsAutoModel = question >> LLMRun() >> SubstringEvaluator("AutoModelForCausalLM")
  8. if __name__ == "__main__":
  9. print(run_test(TestWhatIsAutoModel))