my-speech / entrypoint.sh
joytou's picture
Disabled cuda
07331e0
raw
history blame contribute delete
188 Bytes
#!/bin/bash
CUDA_ENABLED="false"
CUDA_ENABLED=${CUDA_ENABLED:-true}
DEVICE=""
if [ "${CUDA_ENABLED}" != "true" ]; then
DEVICE="--device cpu"
fi
exec python tools/webui.py ${DEVICE}