Spaces:
Running
Running
whisper : update to ggml-backend changes (#0)
Browse files- src/whisper.cpp +1 -1
src/whisper.cpp
CHANGED
|
@@ -597,7 +597,7 @@ static bool whisper_sched_graph_init(struct whisper_sched & allocr, std::vector<
|
|
| 597 |
auto & sched = allocr.sched;
|
| 598 |
auto & meta = allocr.meta;
|
| 599 |
|
| 600 |
-
sched = ggml_backend_sched_new(backends.data(), nullptr, backends.size(), WHISPER_MAX_NODES, false);
|
| 601 |
|
| 602 |
meta.resize(ggml_tensor_overhead()*WHISPER_MAX_NODES + ggml_graph_overhead());
|
| 603 |
|
|
|
|
| 597 |
auto & sched = allocr.sched;
|
| 598 |
auto & meta = allocr.meta;
|
| 599 |
|
| 600 |
+
sched = ggml_backend_sched_new(backends.data(), nullptr, backends.size(), WHISPER_MAX_NODES, false, true);
|
| 601 |
|
| 602 |
meta.resize(ggml_tensor_overhead()*WHISPER_MAX_NODES + ggml_graph_overhead());
|
| 603 |
|