소스 검색

feat(model): use specific CPU backend initialization

沉默の金 5 달 전
부모
커밋
24c65ffeec
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  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();
     }