fix

Signed-off-by: Alexandros Koumparoulis <[email protected]>

Author: akoumpa

nemo_automodel/components/_peft/lora.py

@@ -341,7 +341,10 @@ def apply_lora_to_linear_modules(
         w.requires_grad_(False)
 
     matcher = ModuleMatcher(
-        peft_config.target_modules, peft_config.exclude_modules, peft_config.match_all_linear, model,
+        peft_config.target_modules,
+        peft_config.exclude_modules,
+        peft_config.match_all_linear,
+        model,
     )
     num_modules_matched = 0
     for name, module in list(model.named_modules()):

nemo_automodel/components/_peft/module_matcher.py

@@ -36,6 +36,7 @@ def wildcard_match(pattern, key):
     match = regex_pattern.match(key)
     return match is not None
 
+
 def _get_model_embedding_ptrs(model: nn.Module) -> list[str]:
     ptrs = []
     for name, module in model.named_modules():
@@ -44,6 +45,7 @@ def _get_model_embedding_ptrs(model: nn.Module) -> list[str]:
         ptrs.append(module.weight.data_ptr())
     return ptrs
 
+
 def _get_tied_target_modules(model: nn.Module) -> list[str]:
     if model is None:
         return []