tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

fix rebase issues

Browse Source
This commit is contained in:
Wing Lian
2025-06-05 15:31:28 -07:00
parent cfcd69df0d
commit 01382b9a79
2 changed files with 1 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/axolotl/core/builders/causal.py
View File

@@ -21,11 +21,6 @@ from axolotl.core.trainers import (
AxolotlTrainer, AxolotlTrainer,
ReLoRATrainer, ReLoRATrainer,
) )
from axolotl.core.training_args import (
AxolotlPRMConfig,
AxolotlRewardConfig,
AxolotlTrainingArguments,
)
from axolotl.integrations.base import PluginManager from axolotl.integrations.base import PluginManager
from axolotl.monkeypatch.multipack import SUPPORTED_MULTIPACK_MODEL_TYPES from axolotl.monkeypatch.multipack import SUPPORTED_MULTIPACK_MODEL_TYPES
from axolotl.monkeypatch.relora import ReLoRACallback from axolotl.monkeypatch.relora import ReLoRACallback

2
src/axolotl/utils/trainer.py
View File

@@ -628,7 +628,7 @@ def setup_trainer(
A trainer instance (either `HFRLTrainer` or `HFCausalTrainer`) configured based A trainer instance (either `HFRLTrainer` or `HFCausalTrainer`) configured based
on the provided parameters. on the provided parameters.
""" """
from axolotl.core.trainer_builder import HFCausalTrainerBuilder, HFRLTrainerBuilder from axolotl.core.builders import HFCausalTrainerBuilder, HFRLTrainerBuilder
if ( if (
cfg.torch_compile cfg.torch_compile