Skip to content

Commit 40fd72a

Browse files
Update peft_finetuning.ipynb (meta-llama#461)
2 parents 0e09458 + fda8482 commit 40fd72a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

recipes/finetuning/huggingface_trainer/peft_finetuning.ipynb

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -251,7 +251,7 @@
251251
" get_peft_model,\n",
252252
" LoraConfig,\n",
253253
" TaskType,\n",
254-
" prepare_model_for_int8_training,\n",
254+
" prepare_model_for_kbit_training,\n",
255255
" )\n",
256256
"\n",
257257
" peft_config = LoraConfig(\n",
@@ -264,7 +264,7 @@
264264
" )\n",
265265
"\n",
266266
" # prepare int-8 model for training\n",
267-
" model = prepare_model_for_int8_training(model)\n",
267+
" model = prepare_model_for_kbit_training(model)\n",
268268
" model = get_peft_model(model, peft_config)\n",
269269
" model.print_trainable_parameters()\n",
270270
" return model, peft_config\n",

0 commit comments

Comments
 (0)