|
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