fix(peft): adapt to latest internal API

huggingface · Jan 8, 2025 · 408d3b6 · 408d3b6
1 parent 6639b1d
commit 408d3b6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/optimum/neuron/distributed/utils.py b/optimum/neuron/distributed/utils.py
@@ -412,6 +412,7 @@ def _peft_tuner_embedding_to_parallel_embedding(
                     config.init_lora_weights,
                     config.use_rslora,
                     config.use_dora,
+                    config.lora_bias,
                 )
                 mark_parameter_init_status_during_parallelization(parent.lora_embedding_A[adapter_name], True)
                 mark_parameter_init_status_during_parallelization(parent.lora_embedding_B[adapter_name], True)

diff --git a/setup.py b/setup.py
@@ -32,7 +32,7 @@
     "diffusers>=0.28.0, <=0.30.3",
     "safetensors",
     "sentence-transformers >= 2.2.0",
-    "peft",
+    "peft>=0.14.0",
     "trl==0.11.4",
     "compel",
     "rjieba",
@@ -72,7 +72,7 @@
         "neuronx_distributed==0.9.0",
         "libneuronxla==2.0.5347.0",
     ],
-    "diffusers": ["diffusers>=0.28.0, <=0.30.3", "peft"],
+    "diffusers": ["diffusers>=0.28.0, <=0.30.3", "peft>=0.14.0"],
     "sentence-transformers": ["sentence-transformers >= 2.2.0"],
 }