build: update dependencies for reforge and fix gradio-allowed-path fo…

…r gradio 4

- Update PyTorch, TorchVision, and Xformers to newer versions in Dockerfile
- Add `--gradio-allowed-path "."` argument to the `launch.py` command in entrypoint.sh

Signed-off-by: 陳鈞 <jim60105@gmail.com>

Dockerfile

@@ -60,9 +60,9 @@ RUN --mount=type=cache,id=pip-$TARGETARCH$TARGETVARIANT,sharing=locked,target=/r
     pip install -U --force-reinstall pip setuptools==69.5.1 wheel && \
     pip install -U --extra-index-url https://download.pytorch.org/whl/cu121 --extra-index-url https://pypi.nvidia.com \
     # `torch` (3.6G) and the underlying package `triton` (276M), `torchvision` is small but install together
-    torch==2.1.2 torchvision==0.16.2 \
+    torch==2.3.1 torchvision==0.18.1 \
     # `xformers` (471M)
-    xformers==0.0.23.post1
+    xformers==0.0.27
 
 # Install requirements
 RUN --mount=type=cache,id=pip-$TARGETARCH$TARGETVARIANT,sharing=locked,target=/root/.cache/pip \

entrypoint.sh

@@ -35,7 +35,7 @@ install_requirements() {
             --extra-index-url https://download.pytorch.org/whl/cu121 \
             --extra-index-url https://pypi.nvidia.com \
             -r requirements_versions.txt \
-            torch==2.1.2 torchvision==0.16.2 xformers==0.0.23.post1
+            torch==2.3.1 torchvision==0.18.1 xformers==0.0.27
         pip cache purge
 
         if [ "$(uname -m)" = "x86_64" ]; then
@@ -67,7 +67,7 @@ install_requirements
 trap handle_sigint INT
 
 echo "Starting WebUI with arguments: $*"
-python3 /app/launch.py --listen --port 7860 --data-dir /data "$@" &
+python3 /app/launch.py --listen --port 7860 --data-dir /data --gradio-allowed-path "." "$@" &
 python_pid=$!
 wait $python_pid