ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update trainer.py

Browse source
This commit is contained in:
Daniel Han 2024-11-21 01:49:46 -08:00
commit 8120bb875e
1 changed files with 7 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

8
unsloth/trainer.py
View file

@ -21,7 +21,12 @@ import trl
import inspect
from trl import SFTTrainer
from . import is_bfloat16_supported
from unsloth_zoo.training_utils import unsloth_train as _unsloth_train
from unsloth_zoo.training_utils import (
unsloth_train as _unsloth_train,
)
from unsloth_zoo.vision_utils import (
UnslothVisionDataCollator,
)
from packaging.version import Version
import dataclasses
@ -30,6 +35,7 @@ __all__ = [
"UnslothTrainer",
"unsloth_train",
"_patch_trl_trainer",
"UnslothVisionDataCollator",
]
# Unsloth gradient accumulation fix: