From 2303238e44e68b7690dd8b34be2176e8c046c42c Mon Sep 17 00:00:00 2001
From: Chris McMaster <mcmastc1@gmail.com>
Date: Wed, 7 Feb 2024 11:16:46 +1100
Subject: [PATCH] Update olmo.py (#419)

exit should be imported outside of interactive mode
---
 llms/mlx_lm/models/olmo.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/llms/mlx_lm/models/olmo.py b/llms/mlx_lm/models/olmo.py
index d4df3839..95dec5c6 100644
--- a/llms/mlx_lm/models/olmo.py
+++ b/llms/mlx_lm/models/olmo.py
@@ -6,6 +6,8 @@ import mlx.nn as nn
 
 from .base import BaseModelArgs
 
+from sys import exit
+
 try:
     import hf_olmo
 except ImportError: