Skip to content

Commit

Permalink
Fix CIs
Browse files Browse the repository at this point in the history
  • Loading branch information
skyline75489 committed Dec 17, 2024
1 parent ed1fb47 commit 2d6a8ca
Show file tree
Hide file tree
Showing 3 changed files with 4 additions and 6 deletions.
4 changes: 2 additions & 2 deletions src/java/src/main/java/ai/onnxruntime/genai/Config.java
Original file line number Diff line number Diff line change
Expand Up @@ -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;
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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;

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 2d6a8ca

Please sign in to comment.