From fb9775703f96bebf08db6d944879294ce2459916 Mon Sep 17 00:00:00 2001 From: Kassim Benhaddad Date: Tue, 7 Jan 2025 17:57:11 +0000 Subject: [PATCH] doc: upate Docker README to show new config option --- docker/README.md | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/docker/README.md b/docker/README.md index f3dde85..ccbc1c6 100644 --- a/docker/README.md +++ b/docker/README.md @@ -16,13 +16,19 @@ Run docker image with a local model cache (so it is fast when container is start mkdir -p $HOME/models/ export MODEL_DIR=$HOME/models/ -docker run -it -p 43839:43839 --platform=linux/amd64 --gpus all -v $MODEL_DIR:/workspace/instantmesh/models instantmesh +docker run -it -p 43839:43839 --platform=linux/amd64 --gpus all -v $MODEL_DIR:/workspace/instantmesh/ckpts/models instantmesh ``` To use specific GPUs: ```bash -docker run -it -p 43839:43839 --platform=linux/amd64 --gpus '"device=0,1"' -v $MODEL_DIR:/workspace/instantmesh/models instantmesh +docker run -it -p 43839:43839 --platform=linux/amd64 --gpus '"device=0,1"' -v $MODEL_DIR:/workspace/instantmesh/ckpts instantmesh +``` + +To use a different model file: + +```bash +docker run -it -p 43839:43839 --platform=linux/amd64 --gpus all -v $MODEL_DIR:/workspace/instantmesh/ckpts -e CONFIG=instant-mesh-base instantmesh ``` Navigate to `http://localhost:43839` to use the demo.