Update bark.cpp

PABannier · Dec 15, 2024 · 5184ef4 · 5184ef4
1 parent 5d5be84
commit 5184ef4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bark.cpp b/bark.cpp
@@ -823,7 +823,7 @@ static bool bark_model_load(std::ifstream & fin,
 #ifdef GGML_USE_CUDA
     if (n_gpu_layers > 0) {
         fprintf(stderr, "%s: using CUDA backend\n", __func__);
-        model.backend = ggml_backend_cuda_init();
+        model.backend = ggml_backend_cuda_init(0);
         if (!model.backend) {
             fprintf(stderr, "%s: ggml_backend_cuda_init() failed\n", __func__);
         }