From 2d6a8ca1418c4d047b14d3949393e21e3e7b43a6 Mon Sep 17 00:00:00 2001 From: Chester Liu <4710575+skyline75489@users.noreply.github.com> Date: Tue, 17 Dec 2024 08:50:51 +0000 Subject: [PATCH] Fix CIs --- src/java/src/main/java/ai/onnxruntime/genai/Config.java | 4 ++-- src/java/src/main/java/ai/onnxruntime/genai/NamedTensors.java | 4 +--- .../ai/onnxruntime_genai/example/javavalidator/SimpleTest.kt | 2 +- 3 files changed, 4 insertions(+), 6 deletions(-) diff --git a/src/java/src/main/java/ai/onnxruntime/genai/Config.java b/src/java/src/main/java/ai/onnxruntime/genai/Config.java index 25d1f052b..8b11832f7 100644 --- a/src/java/src/main/java/ai/onnxruntime/genai/Config.java +++ b/src/java/src/main/java/ai/onnxruntime/genai/Config.java @@ -4,8 +4,8 @@ package ai.onnxruntime.genai; /** - * Use Config to set the ORT execution providers (EPs) and their options. The EPs are applied based on - * insertion order. + * Use Config to set the ORT execution providers (EPs) and their options. The EPs are applied based + * on insertion order. */ public final class Config implements AutoCloseable { private long nativeHandle; diff --git a/src/java/src/main/java/ai/onnxruntime/genai/NamedTensors.java b/src/java/src/main/java/ai/onnxruntime/genai/NamedTensors.java index b2b2f686f..d2b2e2224 100644 --- a/src/java/src/main/java/ai/onnxruntime/genai/NamedTensors.java +++ b/src/java/src/main/java/ai/onnxruntime/genai/NamedTensors.java @@ -3,9 +3,7 @@ */ package ai.onnxruntime.genai; -/** - * This class is a list of tensors with names that match up with model input names. - */ +/** This class is a list of tensors with names that match up with model input names. */ public class NamedTensors implements AutoCloseable { private long nativeHandle; diff --git a/src/java/src/test/android/app/src/androidTest/java/ai/onnxruntime_genai/example/javavalidator/SimpleTest.kt b/src/java/src/test/android/app/src/androidTest/java/ai/onnxruntime_genai/example/javavalidator/SimpleTest.kt index 6c8d9705c..259409875 100644 --- a/src/java/src/test/android/app/src/androidTest/java/ai/onnxruntime_genai/example/javavalidator/SimpleTest.kt +++ b/src/java/src/test/android/app/src/androidTest/java/ai/onnxruntime_genai/example/javavalidator/SimpleTest.kt @@ -90,7 +90,7 @@ class SimpleTest { // the test model requires manual input as the token ids have to be < 1000 but the configured tokenizer // has a larger vocab size and the input ids it generates are not valid. val model = Model(newModelPath) - val params = model.createGeneratorParams() + val params = GeneratorParams(model) val sequenceLength = 4 val batchSize = 2