From 4fdc88e9e1698012ba03d8c761a7057045d6cc39 Mon Sep 17 00:00:00 2001 From: Will Miao <13051207myq@gmail.com> Date: Wed, 16 Apr 2025 22:19:38 +0800 Subject: [PATCH] feat: Enhance LoraLoaderExtractor to extract base filename from lora_name input --- py/metadata_collector/node_extractors.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/py/metadata_collector/node_extractors.py b/py/metadata_collector/node_extractors.py index 081dd8c1..612059d6 100644 --- a/py/metadata_collector/node_extractors.py +++ b/py/metadata_collector/node_extractors.py @@ -90,6 +90,8 @@ class LoraLoaderExtractor(NodeMetadataExtractor): return lora_name = inputs.get("lora_name") + # Extract base filename without extension from path + lora_name = os.path.splitext(os.path.basename(lora_name))[0] strength_model = round(float(inputs.get("strength_model", 1.0)), 2) # Use the standardized format with lora_list