From 17ff289073140637258d815f347939812a5f2874 Mon Sep 17 00:00:00 2001
From: Thomas Gauthier-Caron <thomas.gauthier.c@gmail.com>
Date: Thu, 18 Jul 2024 15:34:59 -0400
Subject: [PATCH] [mergekit-extract-lora] renamed a method for clarity

---
 mergekit/scripts/extract_lora.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mergekit/scripts/extract_lora.py b/mergekit/scripts/extract_lora.py
index 7db59743..ff063232 100644
--- 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,