Skip to content

Commit 2ce84a7

Browse files
authored
Merge pull request #4 from NimbleEdge/jpuneet/call-tokenizer-is_loaded
Call tokenizer_->is_loaded
2 parents 4e9eaad + 7c0b62f commit 2ce84a7

3 files changed

Lines changed: 3 additions & 3 deletions

File tree

‎.gitmodules‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@
3030
url = https://github.com/Maratyszcza/pthreadpool.git
3131
[submodule "extension/llm/tokenizers"]
3232
path = extension/llm/tokenizers
33-
url = https://github.com/pytorch-labs/tokenizers.git
33+
url = https://github.com/NimbleEdge/tokenizers.git
3434
[submodule "kernels/optimized/third-party/eigen"]
3535
path = kernels/optimized/third-party/eigen
3636
url = https://gitlab.com/libeigen/eigen.git

‎extension/llm/runner/text_token_generator.h‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,7 @@ class ET_EXPERIMENTAL TextTokenGenerator {
151151
*/
152152
bool inline is_loaded() const {
153153
// Implementation to check if resources are loaded
154-
return text_decoder_runner_->is_method_loaded();
154+
return tokenizer_->is_loaded() && text_decoder_runner_->is_method_loaded();
155155
}
156156

157157
private:

‎extension/llm/tokenizers‎

0 commit comments

Comments
 (0)