From f058ff6ccd0acb26c5ae4a704d472c1a3a3bb16d Mon Sep 17 00:00:00 2001
From: Hamid Shojanazeri <hamid.nazeri2010@gmail.com>
Date: Thu, 21 Mar 2024 15:31:15 +0000
Subject: [PATCH] update due to peft new release
---
src/llama_recipes/finetuning.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/llama_recipes/finetuning.py b/src/llama_recipes/finetuning.py
index 6b5650b2..39b974f6 100644
--- a/src/llama_recipes/finetuning.py
+++ b/src/llama_recipes/finetuning.py
@@ -9,7 +9,7 @@ import fire
import random
import torch
import torch.optim as optim
-from peft import get_peft_model, prepare_model_for_int8_training
+from peft import get_peft_model, prepare_model_for_kbit_training
from torch.distributed.fsdp import (
FullyShardedDataParallel as FSDP,
ShardingStrategy
--
GitLab