Rename NuGet -GPU package to -CUDA

This commit is contained in:
Reuben Morais 2020-08-06 16:16:33 +02:00
parent c31df0fd4c
commit 1e8213c385
2 changed files with 2 additions and 2 deletions

View File

@ -11,7 +11,7 @@ bitrate=$1
set_ldc_sample_filename "${bitrate}"
if [ "${package_option}" = "cuda" ]; then
PROJECT_NAME="Mozilla-Voice-STT-GPU"
PROJECT_NAME="Mozilla-Voice-STT-CUDA"
elif [ "${package_option}" = "--tflite" ]; then
PROJECT_NAME="Mozilla-Voice-STT-TFLite"
model_source=${DEEPSPEECH_TEST_MODEL//.pb/.tflite}

View File

@ -16,7 +16,7 @@ BAZEL_TARGETS="
if [ "${package_option}" = "--cuda" ]; then
BAZEL_ENV_FLAGS="TF_NEED_CUDA=1 ${TF_CUDA_FLAGS}"
BAZEL_BUILD_FLAGS="${BAZEL_CUDA_FLAGS} ${BAZEL_EXTRA_FLAGS} ${BAZEL_OPT_FLAGS}"
PROJECT_NAME="Mozilla-Voice-STT-GPU"
PROJECT_NAME="Mozilla-Voice-STT-CUDA"
elif [ "${package_option}" = "--tflite" ]; then
PROJECT_NAME="Mozilla-Voice-STT-TFLite"
BAZEL_BUILD_FLAGS="--define=runtime=tflite ${BAZEL_OPT_FLAGS} ${BAZEL_EXTRA_FLAGS}"