diff --git a/README.md b/README.md
index 09c35e4..04018cc 100644
--- a/README.md
+++ b/README.md
@@ -84,7 +84,7 @@ Usage:
 Options:
   -b BRANCH, --branch=BRANCH    Select which branch of kohya to check out on new installs.
   -d DIR, --dir=DIR             The full path you want kohya_ss installed to.
-  -g, --git_repo                You can optionally provide a git repo to check out for runpod installation. Useful for custom forks.
+  -g REPO, --git_repo=REPO      You can optionally provide a git repo to check out for runpod installation. Useful for custom forks.
   -h, --help                    Show this screen.
   -i, --interactive             Interactively configure accelerate instead of using default config file.
   -n, --no-update               Do not update kohya_ss repo. No git pull or clone operations.
diff --git a/setup.sh b/setup.sh
index 0bf901e..bc98d6e 100755
--- a/setup.sh
+++ b/setup.sh
@@ -19,10 +19,10 @@ Usage:
 Options:
   -b BRANCH, --branch=BRANCH    Select which branch of kohya to check out on new installs.
   -d DIR, --dir=DIR             The full path you want kohya_ss installed to.
-  -g, --git_repo                You can optionally provide a git repo to check out for runpod installation. Useful for custom forks.
+  -g REPO, --git_repo=REPO      You can optionally provide a git repo to check out for runpod installation. Useful for custom forks.
   -h, --help                    Show this screen.
   -i, --interactive             Interactively configure accelerate instead of using default config file.
-  -n, --no-git-update               Do not update kohya_ss repo. No git pull or clone operations.
+  -n, --no-git-update           Do not update kohya_ss repo. No git pull or clone operations.
   -p, --public                  Expose public URL in runpod mode. Won't have an effect in other modes.
   -r, --runpod                  Forces a runpod installation. Useful if detection fails for any reason.
   -s, --skip-space-check        Skip the 10Gb minimum storage space check.