Skip to content

Commit 65471c7

Browse files
committed
Register OpenLM models at startup
1 parent ce5cea9 commit 65471c7

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

eval/eval.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,11 @@
3131
from eval.eval_tracker import DCEvaluationTracker
3232
from eval.task import TaskManager as InstructTaskManager
3333

34+
# Register OpenLM/DCLM models with HuggingFace Transformers
35+
try:
36+
import open_lm.hf
37+
except ImportError:
38+
pass
3439

3540
_BIT_CAP = 15_000
3641

0 commit comments

Comments
 (0)