update

Author: rohitgr7

README.md

@@ -97,7 +97,20 @@ Personal recommendation is to run this within a docker container.
 
 ## Stable Diffusion
 
+1. Create the image
+
+   ```bash
+   docker build -t triton_cc_pt:0.0.1 -f dockers/Dockerfile.cpu.pt .
+   ```
+
+1. To start a container
+
+   ```bash
+   bash bash_scripts/triton_server_sd.sh
+   ```
+
 1. While compiling Unet with ONNX, it will create multiple files because the model size >2GB.
+
 1. If loading all of these models for a pipeline doesn't work and doesn't show any significant info in the logs, try loading them individually with `--log-verbose=1`.
 
 ## Features

bash_scripts/triton_server_tensorflow.sh

@@ -0,0 +1 @@
+docker run -p 8000:8000 -p 8001:8001 -p 8002:8002 --rm -it -v ${PWD}/models_tf/:/project/models_tf/ -v ${PWD}/weights_tf/:/project/weights_tf/ triton_cc_tf:0.0.1 tritonserver --model-repository models_tf/ --log-verbose=1 --model-control-mode=poll