Handle patch application fail better

run_scripts/set_up.bash

@@ -44,7 +44,10 @@ python -m pip install --upgrade pip
 cd "$MEGATRON_DEEPSPEED_REPO"
 ((DO_PULL)) && git stash && git pull --rebase origin main && git stash pop || :
 git am "$OLDPWD"/../0001-Build-fused-kernels-in-temporary-directory.patch \
-    || echo 'WARNING: Could not apply patch for building in temp dir.'
+    || (
+    git am --abort;
+    echo 'WARNING: Could not apply patch for building in temp dir.'
+)
 # Remove previous build if any.
 rm -rf megatron/fused_kernels/build