diff --git a/unsloth/trainer.py b/unsloth/trainer.py index d26389424c..01b8822bd5 100644 --- a/unsloth/trainer.py +++ b/unsloth/trainer.py @@ -23,11 +23,8 @@ from functools import wraps import trl import inspect from trl import SFTTrainer -# Import via the concrete _utils module instead of the partially-initialized -# `unsloth` namespace. The MLX-aware `unsloth/__init__.py` loads the GPU -# surface via `from ._gpu_init import *`, which transitively triggers -# `from .trainer import *` BEFORE the star export has propagated back to -# `unsloth`, so `from . import is_bfloat16_supported` raises at import time. + +# why: bypass partially-initialised unsloth ns during _gpu_init load from .models._utils import is_bfloat16_supported from unsloth.utils import ( configure_padding_free,