From 73cd16b7b5d8b2160e99001f4febccec536d9a90 Mon Sep 17 00:00:00 2001 From: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com> Date: Wed, 8 Feb 2023 18:43:00 +0530 Subject: [PATCH] quality --- src/peft/peft_model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/peft/peft_model.py b/src/peft/peft_model.py index b6a6628..49bfce1 100644 --- a/src/peft/peft_model.py +++ b/src/peft/peft_model.py @@ -19,8 +19,8 @@ import warnings import torch from accelerate import dispatch_model, infer_auto_device_map -from accelerate.utils import get_balanced_memory from accelerate.hooks import AlignDevicesHook, add_hook_to_module, remove_hook_from_submodules +from accelerate.utils import get_balanced_memory from torch.nn import BCEWithLogitsLoss, CrossEntropyLoss, MSELoss from transformers import PreTrainedModel from transformers.modeling_outputs import SequenceClassifierOutput, TokenClassifierOutput