Skip to content

Commit cd63ef0

Browse files
authored
Fix unchecked Result in get_llm_metadata()
Differential Revision: D99707695 Pull Request resolved: #18749
1 parent 71db7c7 commit cd63ef0

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

extension/llm/runner/llm_runner_helper.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -121,8 +121,9 @@ get_llm_metadata(tokenizers::Tokenizer* tokenizer, Module* module) {
121121
auto& value = pair.second;
122122

123123
if (method_names.count(method_name)) {
124-
auto get_result = module->get(method_name);
125-
value = get_result.get().toScalar().to<decltype(metadata)::mapped_type>();
124+
value = ET_UNWRAP(module->get(method_name))
125+
.toScalar()
126+
.to<decltype(metadata)::mapped_type>();
126127
} else {
127128
ET_LOG(
128129
Info,

0 commit comments

Comments
 (0)