entrypoint.sh 167 B

12345678910
  1. #!/bin/bash
  2. CUDA_ENABLED=${CUDA_ENABLED:-true}
  3. DEVICE=""
  4. if [ "${CUDA_ENABLED}" != "true" ]; then
  5. DEVICE="--device cpu"
  6. fi
  7. exec python tools/webui.py ${DEVICE}