Refactor pipeline definition for finetuning embeddings

zenml-io · Nov 12, 2024 · a2670f7 · a2670f7
1 parent 499a7b1
commit a2670f7
Showing 1 changed file with 2 additions and 8 deletions.
diff --git 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)