From cfa80dace02883c4f6d9aa49218e6f7ef1b65a7b Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 16 Dec 2024 14:27:26 -0500 Subject: [PATCH] import typo --- src/axolotl/integrations/liger/trainer/dpo_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/axolotl/integrations/liger/trainer/dpo_trainer.py b/src/axolotl/integrations/liger/trainer/dpo_trainer.py index 06be574b1..6381533bd 100644 --- a/src/axolotl/integrations/liger/trainer/dpo_trainer.py +++ b/src/axolotl/integrations/liger/trainer/dpo_trainer.py @@ -4,8 +4,8 @@ integration of liger dpo kernels with dpotrainer from typing import Dict, List, Literal, Union import torch -from liger_kernal.transformers.trainer.orpo_trainer import _FSDPForwardRedirection from liger_kernel.chunked_loss import LigerFusedLinearDPOLoss +from liger_kernel.transformers.trainer.orpo_trainer import _FSDPForwardRedirection from torch import nn from torch.distributed.fsdp import FullyShardedDataParallel