Spaces:

PEFT
/

causal-language-modeling

Running

Web333 commited on Apr 30

Commit

a9a35ef

•

1 Parent(s): 1c735d3

`prepare_model_for_int8_training` has been deprecated for quite some time, with PEFT v0.10.0, it has been removed. Please use `prepare_model_for_kbit_training` instead.

Files changed (1) hide show

lora_clm_with_additional_tokens.ipynb CHANGED Viewed

@@ -41,8 +41,8 @@
     "    PeftConfig,\n",
     "    PeftModel,\n",
     "    get_peft_model,\n",
-    "    prepare_model_for_int8_training,\n",
     ")\n",
     "from transformers import (\n",
     "    AutoModelForCausalLM,\n",
     "    AutoTokenizer,\n",

     "    PeftConfig,\n",
     "    PeftModel,\n",
     "    get_peft_model,\n",
     ")\n",
+    "from peft import prepare_model_for_kbit_training as prepare_model_for_int8_training\n",
     "from transformers import (\n",
     "    AutoModelForCausalLM,\n",
     "    AutoTokenizer,\n",