Skip to content

Commit

Permalink
Merge pull request #239 from MichaelClifford/fix-238
Browse files Browse the repository at this point in the history
set caching to False for final_eval_task
  • Loading branch information
tumido authored Dec 17, 2024
2 parents f82864d + 80756a9 commit cd8bce7
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
1 change: 1 addition & 0 deletions pipeline.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down
3 changes: 1 addition & 2 deletions pipeline.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit cd8bce7

Please sign in to comment.