PEFT modules resolution: corrected warning

gaussalgo · Apr 18, 2024 · c7f35e2 · c7f35e2
1 parent 53979f2
commit c7f35e2
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/adaptor/lang_module.py b/adaptor/lang_module.py
@@ -87,8 +87,7 @@ def load_head(self,
             else:
                 PeftModelCls = HEAD_TO_MODEL_CLS[head_type]["peft"]
                 # if that fails, trying to load as a PEFT model
-                logger.warning("Loading model_name_or_path='%s' as full transformer failed. "
-                               "Attempting to load it as peft model.", model_name_or_path)
+                logger.warning("Loading model_name_or_path='%s' as peft model.", model_name_or_path)
                 # base model resolution
                 # we avoid reloading the base model separately for each lora module