@@ -442,12 +442,12 @@ def gen_standalone():
442
442
443
443
# The list of executor names to extract details from to generate the standalone script
444
444
executors = {
445
- "exec-data-processing-op" : 'data_processing_op(max_seq_len=4096 , max_batch_len=20000 , sdg="/data/data ", model="/data/model ", processed_data="/data/processed_data ")' ,
446
- "exec-sdg-op" : 'sdg_op(num_instructions_to_generate=2 , repo_branch="", repo_pr="" , taxonomy="/data/taxonomy ", sdg="/data/generated ")' ,
445
+ "exec-data-processing-op" : 'data_processing_op(max_seq_len={MAX_SEQ_LEN} , max_batch_len={MAX_BATCH_LEN} , sdg="{DATA_PVC_SDG_PATH} ", model="{DATA_PVC_SDG_PATH} ", processed_data="{PREPROCESSED_DATA_PATH} ")' ,
446
+ "exec-sdg-op" : 'sdg_op(num_instructions_to_generate={num_instructions_to_generate} , repo_branch="{exec_git_clone_op_repo_branch} ", repo_pr={exec_git_clone_op_repo_pr} , taxonomy="{TAXONOMY_DATA_PATH} ", sdg="{SDG_GENERATED_DATA_PATH} ")' ,
447
447
"exec-git-clone-op" : {},
448
- "exec-huggingface-importer-op" : 'huggingface_importer_op(repo_name="ibm-granite/granite-7b-base ", model="/data/model ")' ,
449
- "exec-run-mt-bench-op" : 'run_mt_bench_op(best_score_file="/data/mt-bench-best.txt ",mt_bench_output="/data/mt-bench-results.txt ", models_folder="/data/model/output/phase_2/hf_format ", models_path_prefix="/data/model/output/phase_2/hf_format ", max_workers="auto ", merge_system_user_message=False )' ,
450
- "exec-run-final-eval-op" : " run_final_eval_op(mmlu_branch_output='/data/mmlu-branch-best.txt', mt_bench_branch_output='/data/mt-bench-branch-best.txt', candidate_model='/data/model/output/phase_2/hf_format/candidate_model' , taxonomy='/data/taxonomy' , tasks='/data/data' , base_branch='' , candidate_branch='' , device=None, base_model_dir='/data/model' , max_workers='auto' , merge_system_user_message=False , model_dtype='bfloat16' , few_shots=5 , batch_size=8)" ,
448
+ "exec-huggingface-importer-op" : 'huggingface_importer_op(repo_name="{REPO_GRANITE_7B_IMAGE} ", model="{DATA_PVC_MODEL_PATH} ")' ,
449
+ "exec-run-mt-bench-op" : 'run_mt_bench_op(best_score_file="{MT_BENCH_SCORES_PATH} ",mt_bench_output="{MT_BENCH_OUTPUT_PATH} ", models_folder="{CANDIDATE_MODEL_PATH} ", models_path_prefix="{CANDIDATE_MODEL_PATH} ", max_workers="{MAX_WORKERS} ", merge_system_user_message={MERGE_SYSTEM_USER_MESSAGE} )' ,
450
+ "exec-run-final-eval-op" : ' run_final_eval_op(mmlu_branch_output="{MMLU_BRANCH_SCORES_PATH}", mt_bench_branch_output="{MT_BENCH_OUTPUT_PATH}", candidate_model="{CANDIDATE_MODEL_PATH}" , taxonomy="{TAXONOMY_PATH}" , tasks="{DATA_PVC_SDG_PATH}" , base_branch="" , candidate_branch="" , device=None, base_model_dir="{DATA_PVC_MODEL_PATH}" , max_workers="{MAX_WORKERS}" , merge_system_user_message={MERGE_SYSTEM_USER_MESSAGE} , model_dtype="{MODEL_DTYPE}" , few_shots={FEW_SHOTS} , batch_size={BATCH_SIZE})' ,
451
451
}
452
452
453
453
details = {}
0 commit comments