• Joseph Young's avatar
    Updated Dockerfile for CUDA backend · c5948d3e
    Joseph Young authored
    Enabled NVIDIA CUDA backend build stage in the Dockerfile for enhanced performance with GPU support. Moved the environment variable defining the device type for the embedding and TTS models to be shared between CPU and GPU configurations. The default device type for CPU build is now explicitly set to "cpu", while the CUDA build retains "cuda", ensuring clarity and performance optimization across different hardware setups.
    c5948d3e
Dockerfile-cuda 3.79 KB