From 95212edf1f15909edd175eb0825da810c2a7df92 Mon Sep 17 00:00:00 2001
From: oobabooga <112222186+oobabooga@users.noreply.github.com>
Date: Sun, 25 Jun 2023 12:13:15 -0300
Subject: [PATCH] Update training.py

---
 modules/training.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/training.py b/modules/training.py
index d0018a0f..e629d86e 100644
--- a/modules/training.py
+++ b/modules/training.py
@@ -14,7 +14,7 @@ from datasets import Dataset, load_dataset
 from peft import (
     LoraConfig,
     get_peft_model,
-    prepare_model_for_kbit_training,
+    prepare_model_for_int8_training,
     set_peft_model_state_dict
 )