[BugFix] : Small updates to merge component (#3869)

Co-authored-by: Matthias Blondeel <mablonde@microsoft.com>
Azure · Feb 26, 2025 · 06a2ec0 · 06a2ec0
1 parent d74da53
commit 06a2ec0
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/...ng/finetune_acft_image/components/model_converters/medimage_embed_adapter_merge/spec.yaml b/...ng/finetune_acft_image/components/model_converters/medimage_embed_adapter_merge/spec.yaml
@@ -1,7 +1,7 @@
 $schema: https://azuremlschemas.azureedge.net/latest/commandComponent.schema.json
 type: command
 
-version: 0.0.12
+version: 0.0.13
 name: medimage_embedding_adapter_merge
 display_name: Merge Adapter Model with Embedding Generation Model
 description: Import and integrate Adapter and Embedding Generation Model

diff --git a/...training/finetune_acft_image/components/pipeline_components/medimage_insight_ft/spec.yaml b/...training/finetune_acft_image/components/pipeline_components/medimage_insight_ft/spec.yaml
@@ -1,6 +1,6 @@
 $schema: https://azuremlschemas.azureedge.net/latest/pipelineComponent.schema.json
 name: medimage_insight_ft_pipeline
-version: 0.0.14
+version: 0.0.16
 type: pipeline
 display_name: Medical Image Insight Embedding Generator and Classification Adapter Pipeline
 description: Pipeline Component to finetune Hugging Face pretrained models for chat completion task. The component supports optimizations such as LoRA, Deepspeed and ONNXRuntime for performance enhancement. See [docs](https://aka.ms/azureml/components/chat_completion_pipeline) to learn more.
@@ -406,7 +406,7 @@ jobs:
       max_epochs: '${{parent.inputs.max_epochs}}'
   medimage_embedding_adapter_merge:
     type: command
-    component: azureml://registries/mablonde-registry-101/components/medimage_embedding_adapter_merge/versions/0.0.12
+    component: azureml://registries/mablonde-registry-101/components/medimage_embedding_adapter_merge/versions/0.0.13
     compute: '${{parent.inputs.compute_finetune}}'
     resources:
       instance_type: '${{parent.inputs.instance_type_finetune}}'

diff --git a/...tune_acft_image/src/model_converters/medimage_embed_adapter_merge/medimage_model_merge.py b/...tune_acft_image/src/model_converters/medimage_embed_adapter_merge/medimage_model_merge.py
@@ -49,7 +49,7 @@ def merge_models(adapter_model_path, mlflow_model_path, output_dir, hidden_dimen
     logger.info(f"Copied {BEST_METRIC_MODEL} to {artifacts_dir}")
 
     with open(label_file, "r") as f:
-        labels = [l for l in f.read().splitlines() if not l.strip()]
+        labels = [l.strip() for l in f.read().splitlines() if l.strip()]
 
     config = {
         "hidden_dim": hidden_dimensions,