From a2670f7d6f0a5ee8d937703502276f7a0be98998 Mon Sep 17 00:00:00 2001
From: Hamza Tahir <hamza@zenml.io>
Date: Tue, 12 Nov 2024 10:45:54 +0100
Subject: [PATCH] Refactor pipeline definition for finetuning embeddings

---
 llm-complete-guide/pipelines/finetune_embeddings.py | 10 ++--------
 1 file changed, 2 insertions(+), 8 deletions(-)

diff --git a/llm-complete-guide/pipelines/finetune_embeddings.py b/llm-complete-guide/pipelines/finetune_embeddings.py
index 1f30caba..e53ae3f1 100644
--- a/llm-complete-guide/pipelines/finetune_embeddings.py
+++ b/llm-complete-guide/pipelines/finetune_embeddings.py
@@ -20,16 +20,10 @@
     prepare_load_data,
     visualize_results,
 )
-from zenml import Model, pipeline
-from zenml.model.model import ModelStages
-
-model_definition = Model(
-    name=EMBEDDINGS_MODEL_NAME_ZENML,
-    version=ModelStages.LATEST,
-)
+from zenml import pipeline
 
 
-@pipeline(model=model_definition)
+@pipeline
 def finetune_embeddings():
     data = prepare_load_data()
     base_results = evaluate_base_model(dataset=data)