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)