[rust] Download cu124 jni library for cuda (#3327)

deepjavalibrary · Jul 11, 2024 · c6e3cdd · c6e3cdd
1 parent dfdf93b
commit c6e3cdd
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
@@ -38,6 +38,7 @@ public final class LibUtils {
     private static final Pattern VERSION_PATTERN =
             Pattern.compile(
                     "(\\d+\\.\\d+\\.\\d+(-[a-z]+)?)-(\\d+\\.\\d+\\.\\d+)(-SNAPSHOT)?(-\\d+)?");
+    private static final String FLAVOR_CU124 = "cu124";
 
     private static EngineException exception;
 
@@ -106,7 +107,7 @@ private static Path copyJniLibrary(String[] libs) {
             String jniVersion = matcher.group(1);
             String djlVersion = matcher.group(3);
 
-            downloadJniLib(dir, path, djlVersion, jniVersion, classifier, flavor);
+            downloadJniLib(dir, path, djlVersion, jniVersion, classifier, FLAVOR_CU124);
             return dir.toAbsolutePath();
         }