mirror of
https://github.com/unslothai/unsloth.git
synced 2026-05-17 03:56:07 +00:00
Trim trainer.py import-fix comment to one line
This commit is contained in:
parent
e594c38f3f
commit
23550a8e38
1 changed files with 2 additions and 5 deletions
|
|
@ -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,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue