|
7 | 7 | import dev.langchain4j.model.output.Response;
|
8 | 8 | import dev.langchain4j.model.output.TokenUsage;
|
9 | 9 |
|
10 |
| -import java.io.IOException; |
11 | 10 | import java.io.InputStream;
|
12 | 11 | import java.nio.file.Path;
|
13 | 12 | import java.util.ArrayList;
|
|
16 | 15 |
|
17 | 16 | import static dev.langchain4j.internal.Utils.getOrDefault;
|
18 | 17 | import static dev.langchain4j.internal.ValidationUtils.ensureNotEmpty;
|
19 |
| -import static java.nio.file.Files.newInputStream; |
20 | 18 | import static java.util.Collections.singletonList;
|
21 | 19 | import static java.util.concurrent.CompletableFuture.supplyAsync;
|
22 | 20 | import static java.util.concurrent.TimeUnit.SECONDS;
|
@@ -48,19 +46,7 @@ protected static OnnxBertBiEncoder loadFromJar(String modelFileName, String toke
|
48 | 46 | }
|
49 | 47 |
|
50 | 48 | static OnnxBertBiEncoder loadFromFileSystem(Path pathToModel, Path pathToTokenizer, PoolingMode poolingMode) {
|
51 |
| - try { |
52 |
| - return new OnnxBertBiEncoder(newInputStream(pathToModel), newInputStream(pathToTokenizer), poolingMode); |
53 |
| - } catch (IOException e) { |
54 |
| - throw new RuntimeException(e); |
55 |
| - } |
56 |
| - } |
57 |
| - |
58 |
| - static OnnxBertBiEncoder loadFromFileSystem(Path pathToModel, InputStream tokenizer, PoolingMode poolingMode) { |
59 |
| - try { |
60 |
| - return new OnnxBertBiEncoder(newInputStream(pathToModel), tokenizer, poolingMode); |
61 |
| - } catch (IOException e) { |
62 |
| - throw new RuntimeException(e); |
63 |
| - } |
| 49 | + return new OnnxBertBiEncoder(pathToModel, pathToTokenizer, poolingMode); |
64 | 50 | }
|
65 | 51 |
|
66 | 52 | protected abstract OnnxBertBiEncoder model();
|
|
0 commit comments