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 trl
|
||||||
import inspect
|
import inspect
|
||||||
from trl import SFTTrainer
|
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
|
# why: bypass partially-initialised unsloth ns during _gpu_init load
|
||||||
# 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.
|
|
||||||
from .models._utils import is_bfloat16_supported
|
from .models._utils import is_bfloat16_supported
|
||||||
from unsloth.utils import (
|
from unsloth.utils import (
|
||||||
configure_padding_free,
|
configure_padding_free,
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue