Update olmo.py (#419)

exit should be imported outside of interactive mode
This commit is contained in:
Chris McMaster
2024-02-07 11:16:46 +11:00
committed by GitHub
parent 8b77677c05
commit 2303238e44

View File

@@ -6,6 +6,8 @@ import mlx.nn as nn
from .base import BaseModelArgs
from sys import exit
try:
import hf_olmo
except ImportError: