From 80756a9b73e1206af9fd974d8b6d7b41ab1ac704 Mon Sep 17 00:00:00 2001 From: Michael Clifford Date: Tue, 17 Dec 2024 08:41:22 -0500 Subject: [PATCH] set caching to False for final_eval_task Signed-off-by: Michael Clifford --- pipeline.py | 1 + pipeline.yaml | 3 +-- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/pipeline.py b/pipeline.py index 59b9a1f..7f5b0e5 100644 --- a/pipeline.py +++ b/pipeline.py @@ -394,6 +394,7 @@ def pipeline( final_eval_task.after(run_mt_bench_task) final_eval_task.set_accelerator_type("nvidia.com/gpu") final_eval_task.set_accelerator_limit(1) + final_eval_task.set_caching_options(False) output_model_task = pvc_to_model_op( pvc_path="/output/phase_2/model/hf_format/candidate_model", diff --git a/pipeline.yaml b/pipeline.yaml index 482facd..bdcc742 100644 --- a/pipeline.yaml +++ b/pipeline.yaml @@ -1961,8 +1961,7 @@ root: taskInfo: name: pytorchjob-manifest-op-2 run-final-eval-op: - cachingOptions: - enableCache: true + cachingOptions: {} componentRef: name: comp-run-final-eval-op dependentTasks: