Quellcode durchsuchen

feat(model): use specific CPU backend initialization

沉默の金 vor 5 Monaten
Ursprung
Commit
24c65ffeec
1 geänderte Dateien mit 1 neuen und 1 gelöschten Zeilen
  1. 1 1
      src/model.cpp

+ 1 - 1
src/model.cpp

@@ -20,7 +20,7 @@ MelBandRoformer::~MelBandRoformer() {
 void MelBandRoformer::Initialize(const std::string& model_path) {
     // Use best available backend, but allow forcing CPU
     if (std::getenv("MBR_FORCE_CPU")) {
-        backend_ = ggml_backend_cpu_init();
+        backend_ = ggml_backend_init_by_type(GGML_BACKEND_DEVICE_TYPE_CPU, NULL);
     } else {
         backend_ = ggml_backend_init_best();
     }