doc: upate Docker README to show new config option

MrLemur · Jan 7, 2025 · fb97757 · fb97757
1 parent faaf397
commit fb97757
Showing 1 changed file with 8 additions and 2 deletions.
diff --git 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.