added valueerror to mappings

huggingface · Nov 11, 2024 · d3e36eb · d3e36eb
1 parent a86d6f8
commit d3e36eb
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/peft/mapping.py b/src/peft/mapping.py
@@ -189,8 +189,11 @@ def get_peft_model(
         # note: PeftMixedModel does not support autocast_adapter_dtype, so don't pass it
         return PeftMixedModel(model, peft_config, adapter_name=adapter_name)
 
-    if peft_config.task_type not in MODEL_TYPE_TO_PEFT_MODEL_MAPPING.keys() and not peft_config.is_prompt_learning:
-        return PeftModel(model, peft_config, adapter_name=adapter_name, autocast_adapter_dtype=autocast_adapter_dtype)
+    if peft_config.task_type not in MODEL_TYPE_TO_PEFT_MODEL_MAPPING.keys():
+        if not peft_config.is_prompt_learning:
+            return PeftModel(model, peft_config, adapter_name=adapter_name, autocast_adapter_dtype=autocast_adapter_dtype)
+        else:
+            raise ValueError(f"Invalid task type: '{peft_config.task_type}'. Must be one of the following task types: {list(MODEL_TYPE_TO_PEFT_MODEL_MAPPING.keys())}.")
 
     if peft_config.is_prompt_learning:
         peft_config = _prepare_prompt_learning_config(peft_config, model_config)