Update

deepjavalibrary · Jul 11, 2024 · ca0f1ce · ca0f1ce
1 parent 4893401
commit ca0f1ce
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/extensions/tokenizers/src/main/java/ai/djl/huggingface/tokenizers/jni/LibUtils.java b/extensions/tokenizers/src/main/java/ai/djl/huggingface/tokenizers/jni/LibUtils.java
@@ -85,6 +85,7 @@ private static void loadLibrary() {
     private static Path copyJniLibrary(String[] libs) {
         Path cacheDir = Utils.getEngineCacheDir("tokenizers");
         Platform platform = Platform.detectPlatform("tokenizers");
+        String os = platform.getOsPrefix();
         String classifier = platform.getClassifier();
         String flavor = platform.getFlavor();
         String version = platform.getVersion();
@@ -95,10 +96,8 @@ private static Path copyJniLibrary(String[] libs) {
             return dir.toAbsolutePath();
         }
 
-        boolean isCuda = flavor.contains("cu");
-
-        // For cuda, download jni lib files
-        if (isCuda) {
+        // For Linux cuda 12.x, download JNI lib files
+        if (flavor.startsWith("cu12") && !"win".equals(os)) {
             Matcher matcher = VERSION_PATTERN.matcher(version);
             if (!matcher.matches()) {
                 throw new EngineException("Unexpected version: " + version);
@@ -110,7 +109,7 @@ private static Path copyJniLibrary(String[] libs) {
             return dir;
         }
 
-        // For cpu, extract jni lib files from classpath
+        // Extract JNI lib files from classpath
         Path tmp = null;
         try {
             Files.createDirectories(cacheDir);