From 4493d83aea680fe0d220b26f9797b1ba49d9b35f Mon Sep 17 00:00:00 2001
From: mauwii <Mauwii@outlook.de>
Date: Wed, 8 Feb 2023 21:35:19 +0100
Subject: [PATCH] update docker-scripts instructions

---
 docker/build.sh | 17 +++++++++++------
 docker/run.sh   |  3 +--
 2 files changed, 12 insertions(+), 8 deletions(-)

diff --git a/docker/build.sh b/docker/build.sh
index dabbd8378a..7dc98ec208 100755
--- a/docker/build.sh
+++ b/docker/build.sh
@@ -1,12 +1,17 @@
 #!/usr/bin/env bash
 set -e
 
-# How to use: https://invoke-ai.github.io/InvokeAI/installation/INSTALL_DOCKER/#setup
-# Some possible pip extra-index urls (cuda 11.7 is available without extra url):
-#   CUDA 11.6:  https://download.pytorch.org/whl/cu116
-#   ROCm 5.2:   https://download.pytorch.org/whl/rocm5.2
-#   CPU:        https://download.pytorch.org/whl/cpu
-#   as found on https://pytorch.org/get-started/locally/
+# If you want to build a specific flavor, set the CONTAINER_FLAVOR environment variable
+#   e.g. CONTAINER_FLAVOR=cpu ./build.sh
+#   Possible Values are:
+#     - cpu
+#     - cuda
+#     - rocm
+#   Don't forget to also set it when executing run.sh
+#   if it is not set, the script will try to detect the flavor by itself.
+#
+# Doc can be found here:
+#   https://invoke-ai.github.io/InvokeAI/installation/040_INSTALL_DOCKER/
 
 SCRIPTDIR=$(dirname "${BASH_SOURCE[0]}")
 cd "$SCRIPTDIR" || exit 1
diff --git a/docker/run.sh b/docker/run.sh
index ab1ae2011e..eebb533078 100755
--- a/docker/run.sh
+++ b/docker/run.sh
@@ -1,8 +1,7 @@
 #!/usr/bin/env bash
 set -e
 
-# How to use: https://invoke-ai.github.io/InvokeAI/installation/INSTALL_DOCKER/#run-the-container
-# IMPORTANT: You need to have a token on huggingface.co to be able to download the checkpoints!!!
+# How to use: https://invoke-ai.github.io/InvokeAI/installation/040_INSTALL_DOCKER/
 
 SCRIPTDIR=$(dirname "${BASH_SOURCE[0]}")
 cd "$SCRIPTDIR" || exit 1