[mergekit-extract-lora] renamed a method for clarity

arcee-ai · Jul 18, 2024 · 17ff289 · 17ff289
1 parent 0a622ae
commit 17ff289
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mergekit/scripts/extract_lora.py b/mergekit/scripts/extract_lora.py
@@ -211,7 +211,7 @@ def validate_and_combine_details(
     return module_details, base_model_embedding_size, finetuned_model_embedding_size
 
 
-def process_module_details(
+def extract_lora(
     module_details: List[Tuple[str, str]],
     base_model_ref: ModelReference,
     finetuned_model_ref: ModelReference,
@@ -573,7 +573,7 @@ def main(
         extend_vocab,
     )
 
-    lora_weights, ranks = process_module_details(
+    lora_weights, ranks = extract_lora(
         module_details,
         base_model_ref,
         finetuned_model_ref,