Skip to content

Commit 15ffe79

Browse files
committed
Add modulePath setter
1 parent 77ed3f5 commit 15ffe79

File tree

1 file changed

+13
-2
lines changed
  • extension/android/executorch_android/src/main/java/org/pytorch/executorch/extension/llm

1 file changed

+13
-2
lines changed

extension/android/executorch_android/src/main/java/org/pytorch/executorch/extension/llm/LlmModuleConfig.java

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -95,9 +95,20 @@ public static class Builder {
9595
Builder() {}
9696

9797
/**
98-
* Sets the path to the tokenizer resources.
98+
* Sets the path to the module.
9999
*
100-
* @param tokenizerPath Path to tokenizer files/directory
100+
* @param modulePath Path to module
101+
* @return This builder instance for method chaining
102+
*/
103+
public Builder modulePath(String modulePath) {
104+
this.modulePath = modulePath;
105+
return this;
106+
}
107+
108+
/**
109+
* Sets the path to the tokenizer.
110+
*
111+
* @param tokenizerPath Path to tokenizer
101112
* @return This builder instance for method chaining
102113
*/
103114
public Builder tokenizerPath(String tokenizerPath) {

0 commit comments

Comments
 (0)