We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 0e09458 + fda8482 commit 40fd72aCopy full SHA for 40fd72a
recipes/finetuning/huggingface_trainer/peft_finetuning.ipynb
@@ -251,7 +251,7 @@
251
" get_peft_model,\n",
252
" LoraConfig,\n",
253
" TaskType,\n",
254
- " prepare_model_for_int8_training,\n",
+ " prepare_model_for_kbit_training,\n",
255
" )\n",
256
"\n",
257
" peft_config = LoraConfig(\n",
@@ -264,7 +264,7 @@
264
265
266
" # prepare int-8 model for training\n",
267
- " model = prepare_model_for_int8_training(model)\n",
+ " model = prepare_model_for_kbit_training(model)\n",
268
" model = get_peft_model(model, peft_config)\n",
269
" model.print_trainable_parameters()\n",
270
" return model, peft_config\n",
0 commit comments