From 65471c724482540a8d2a8a24c9141b1397000ce1 Mon Sep 17 00:00:00 2001 From: rwh Date: Fri, 27 Jun 2025 14:12:47 +0200 Subject: [PATCH] Register OpenLM models at startup --- eval/eval.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/eval/eval.py b/eval/eval.py index b8563ee5..17cb5d88 100644 --- a/eval/eval.py +++ b/eval/eval.py @@ -31,6 +31,11 @@ from eval.eval_tracker import DCEvaluationTracker from eval.task import TaskManager as InstructTaskManager +# Register OpenLM/DCLM models with HuggingFace Transformers +try: + import open_lm.hf +except ImportError: + pass _BIT_CAP = 15_000