From 5184ef413a6bd9f70110190a2b8b4393e525dfce Mon Sep 17 00:00:00 2001
From: Claude Stabile <claude@stabile.com>
Date: Sun, 15 Dec 2024 12:34:08 +0100
Subject: [PATCH] Update bark.cpp

---
 bark.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/bark.cpp b/bark.cpp
index 6e99d99..6ab0a69 100644
--- 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__);
         }