From af280d17c0d9bd9a3d6650d3e9adeb7720e66361 Mon Sep 17 00:00:00 2001
From: Eleni Verteouri <141945709+EleniVerteouri@users.noreply.github.com>
Date: Fri, 29 Mar 2024 20:21:04 +0100
Subject: [PATCH] Update train_lora.py for kbit training in peft import

---
 fingpt/FinGPT_Benchmark/train_lora.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fingpt/FinGPT_Benchmark/train_lora.py b/fingpt/FinGPT_Benchmark/train_lora.py
index f669fcba..79841f1c 100644
--- a/fingpt/FinGPT_Benchmark/train_lora.py
+++ b/fingpt/FinGPT_Benchmark/train_lora.py
@@ -23,7 +23,7 @@
     LoraConfig,
     get_peft_model,
     get_peft_model_state_dict,
-    prepare_model_for_int8_training,
+    prepare_model_for_kbit_training,
     set_peft_model_state_dict
 )
 from utils import *