Ok can't remove import lm_eval. Move it inside eval condition

larryliu0820 · larryliu0820 · commit d7a4a55336c0 · 2024-09-26T10:31:23.000-07:00
Summary:

Test Plan:

Reviewers:

Subscribers:

Tasks:

Tags:
diff --git a/torchchat.py b/torchchat.py
@@ -82,6 +82,7 @@
 
         generate_main(args)
     elif args.command == "eval":
+        import lm_eval  # noqa
         from torchchat.usages.eval import main as eval_main
 
         eval_main(args)
