Update

deepjavalibrary · Jul 12, 2024 · 9f29e98 · 9f29e98
1 parent 2d67b04
commit 9f29e98
Showing 1 changed file with 2 additions and 1 deletion.
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
@@ -95,9 +95,10 @@ private static Path copyJniLibrary(String[] libs) {
         if (override) {
             logger.info("Uses override TOKENIZERS_FLAVOR: {}", flavor);
         } else {
-            flavor = platform.getFlavor();
             if (Utils.isOfflineMode() || "win".equals(os)) {
                 flavor = "cpu";
+            } else {
+                flavor = platform.getFlavor();
             }
         }