Forráskód Böngészése

feat(model): use specific CPU backend initialization

沉默の金 5 hónapja
szülő
commit
24c65ffeec
1 módosított fájl, 1 hozzáadás és 1 törlés
  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();
     }