|
@@ -20,7 +20,7 @@ MelBandRoformer::~MelBandRoformer() {
|
|
|
void MelBandRoformer::Initialize(const std::string& model_path) {
|
|
void MelBandRoformer::Initialize(const std::string& model_path) {
|
|
|
// Use best available backend, but allow forcing CPU
|
|
// Use best available backend, but allow forcing CPU
|
|
|
if (std::getenv("MBR_FORCE_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 {
|
|
} else {
|
|
|
backend_ = ggml_backend_init_best();
|
|
backend_ = ggml_backend_init_best();
|
|
|
}
|
|
}
|