FROM megatron_image

# 复制 Argument Parser
COPY --from=common_utils parse_torchrun_args.sh /opt/TrainPerfBench/parse_torchrun_args.sh
COPY parse_args.sh /opt/TrainPerfBench/parse_args.sh

# 复制通用脚本
COPY --from=common_utils run_cmd.sh /opt/TrainPerfBench/run_cmd.sh
COPY --from=common_utils check_existence.sh /opt/TrainPerfBench/check_existence.sh
COPY --from=common_utils download_utils.sh /opt/TrainPerfBench/download_utils.sh

# 复制数据下载和训练脚本
COPY download_qwen_datasets.sh /workspace/Megatron-LM/download_qwen_datasets.sh
COPY train_qwen.sh /workspace/Megatron-LM/train_qwen.sh
