From 88e68166b63fca84c3052485f8dbbfa6a4f62539 Mon Sep 17 00:00:00 2001 From: Kangrui Du Date: Mon, 4 May 2026 02:38:44 +0000 Subject: [PATCH] scripts: drop --apply-sgld-monkey-patches from default flowgrpo-aligned launchers The patches incur a small rollout-engine perf hit and are only needed when high-precision rollout/train alignment is required. Default training does not need them; opt-in via the CLI flag for precision-sensitive runs. Co-Authored-By: Claude Opus 4.7 --- scripts/run-diffusion-grpo-ocr-2gpu-flowgrpo-aligned.sh | 1 - scripts/run-diffusion-grpo-ocr-4gpu-flowgrpo-aligned.sh | 1 - scripts/run-diffusion-grpo-pickscore-2gpu-flowgrpo-aligned.sh | 1 - scripts/run-diffusion-grpo-pickscore-4gpu-flowgrpo-aligned.sh | 1 - 4 files changed, 4 deletions(-) diff --git a/scripts/run-diffusion-grpo-ocr-2gpu-flowgrpo-aligned.sh b/scripts/run-diffusion-grpo-ocr-2gpu-flowgrpo-aligned.sh index 947bf4bed3..e0000658ce 100755 --- a/scripts/run-diffusion-grpo-ocr-2gpu-flowgrpo-aligned.sh +++ b/scripts/run-diffusion-grpo-ocr-2gpu-flowgrpo-aligned.sh @@ -91,7 +91,6 @@ hf download --repo-type dataset rockdu/miles-diffusion-datasets \ --diffusion-sde-window-size 2 \ --diffusion-sde-window-range 3,5 \ --diffusion-debug-mode \ - --apply-sgld-monkey-patches \ --diffusion-height 512 \ --diffusion-width 512 \ --save "${SAVE_DIR}" \ diff --git a/scripts/run-diffusion-grpo-ocr-4gpu-flowgrpo-aligned.sh b/scripts/run-diffusion-grpo-ocr-4gpu-flowgrpo-aligned.sh index ee3ffb6640..7950b1eaf1 100755 --- a/scripts/run-diffusion-grpo-ocr-4gpu-flowgrpo-aligned.sh +++ b/scripts/run-diffusion-grpo-ocr-4gpu-flowgrpo-aligned.sh @@ -94,7 +94,6 @@ hf download --repo-type dataset rockdu/miles-diffusion-datasets \ --diffusion-sde-window-size 2 \ --diffusion-sde-window-range 3,5 \ --diffusion-debug-mode \ - --apply-sgld-monkey-patches \ --diffusion-height 512 \ --diffusion-width 512 \ --save "${SAVE_DIR}" \ diff --git a/scripts/run-diffusion-grpo-pickscore-2gpu-flowgrpo-aligned.sh b/scripts/run-diffusion-grpo-pickscore-2gpu-flowgrpo-aligned.sh index 9a344e4ccd..55feaf7413 100755 --- a/scripts/run-diffusion-grpo-pickscore-2gpu-flowgrpo-aligned.sh +++ b/scripts/run-diffusion-grpo-pickscore-2gpu-flowgrpo-aligned.sh @@ -99,7 +99,6 @@ hf download --repo-type dataset rockdu/miles-diffusion-datasets \ --diffusion-sde-window-size 2 \ --diffusion-sde-window-range 3,5 \ --diffusion-debug-mode \ - --apply-sgld-monkey-patches \ --diffusion-height 512 \ --diffusion-width 512 \ --save "${SAVE_DIR}" \ diff --git a/scripts/run-diffusion-grpo-pickscore-4gpu-flowgrpo-aligned.sh b/scripts/run-diffusion-grpo-pickscore-4gpu-flowgrpo-aligned.sh index 908ed91e8f..54ec9a9ef4 100755 --- a/scripts/run-diffusion-grpo-pickscore-4gpu-flowgrpo-aligned.sh +++ b/scripts/run-diffusion-grpo-pickscore-4gpu-flowgrpo-aligned.sh @@ -100,7 +100,6 @@ hf download --repo-type dataset rockdu/miles-diffusion-datasets \ --diffusion-step-strategy-path miles.rollout.step_strategy_hub.sde_window \ --diffusion-sde-window-size 2 \ --diffusion-sde-window-range 3,5 \ - --apply-sgld-monkey-patches \ --diffusion-height 512 \ --diffusion-width 512 \ --save "${SAVE_DIR}" \