Skip to content

Commit f93b497

Browse files
ai-edge-botcopybara-github
authored andcommitted
Internal change
LiteRT-LM-PiperOrigin-RevId: 861936534
1 parent 4753b9d commit f93b497

File tree

3 files changed

+3
-0
lines changed

3 files changed

+3
-0
lines changed

runtime/executor/BUILD

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -119,6 +119,7 @@ cc_library(
119119
"//runtime/components:model_resources_litert_lm",
120120
"//runtime/components:model_resources_task",
121121
"//runtime/components:sentencepiece_tokenizer",
122+
"//runtime/proto:sampler_params_cc_proto",
122123
"//runtime/util:external_file_cc_proto",
123124
"//runtime/util:file_format_util",
124125
"//runtime/util:litert_lm_loader",

runtime/executor/litert_compiled_model_executor_utils.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
#include "litert/cc/litert_tensor_buffer.h" // from @litert
3131
#include "runtime/components/model_resources.h"
3232
#include "runtime/executor/executor_settings_base.h"
33+
#include "runtime/proto/sampler_params.pb.h"
3334

3435
namespace litert::lm {
3536

runtime/executor/llm_litert_compiled_model_executor.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
#include <filesystem> // NOLINT(build/c++17) for std::filesystem::path
2121
#include <memory>
2222
#include <optional>
23+
#include <random>
2324
#include <string>
2425
#include <utility>
2526
#include <variant>

0 commit comments

Comments
 (0)