summaryrefslogtreecommitdiff
path: root/tests/test-tokenizer-0-llama.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test-tokenizer-0-llama.cpp')
-rw-r--r--tests/test-tokenizer-0-llama.cpp7
1 files changed, 7 insertions, 0 deletions
diff --git a/tests/test-tokenizer-0-llama.cpp b/tests/test-tokenizer-0-llama.cpp
index 8630742c..edbd86f8 100644
--- a/tests/test-tokenizer-0-llama.cpp
+++ b/tests/test-tokenizer-0-llama.cpp
@@ -1,5 +1,6 @@
#include "llama.h"
#include "common.h"
+#include "console.h"
#include <cstdio>
#include <string>
@@ -89,6 +90,12 @@ int main(int argc, char **argv) {
return 2;
}
+#ifdef _WIN32
+ // We need this for unicode console support
+ console::init(false, false);
+ atexit([]() { console::cleanup(); });
+#endif
+
bool success = true;
for (const auto & test_kv : k_tests()) {