fix(djl): set onnxengine manually

BakaBotTeam · Oct 3, 2024 · 41d15d5 · 41d15d5
1 parent a9b01f0
commit 41d15d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/ltd/guimc/dlm/DLModel.java b/src/main/java/ltd/guimc/dlm/DLModel.java
@@ -42,7 +42,7 @@ public static void init() {
             Files.copy(modelStream, tempFile.toPath(), StandardCopyOption.REPLACE_EXISTING);
 
             // 使用 DJL (Deep Java Library) 加载 ONNX 模型
-            model = Model.newInstance("Nailong");
+            model = Model.newInstance("Onnx");
             model.load(tempFile.toPath());
 
             // 初始化模型