Skip to content

Commit ffdb0fc

Browse files
ai-edge-botcopybara-github
authored andcommitted
Internal changes only
LiteRT-LM-PiperOrigin-RevId: 893199630
1 parent d9c86eb commit ffdb0fc

File tree

2 files changed

+17
-0
lines changed

2 files changed

+17
-0
lines changed

runtime/executor/executor_settings_base.cc

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -159,6 +159,13 @@ absl::StatusOr<ModelAssets> ModelAssets::Create(
159159
return ModelAssets(std::move(model_file), model_path);
160160
}
161161

162+
// static
163+
absl::StatusOr<ModelAssets> ModelAssets::Create(
164+
std::shared_ptr<litert::lm::MemoryMappedFile> model_file,
165+
absl::string_view model_path) {
166+
return ModelAssets(std::move(model_file), model_path);
167+
}
168+
162169
ModelAssets::ModelAssets(std::shared_ptr<litert::lm::ScopedFile> model_file,
163170
absl::string_view model_path)
164171
: path_(model_path), scoped_file_(std::move(model_file)) {}
@@ -170,6 +177,11 @@ ModelAssets::ModelAssets(
170177
std::shared_ptr<litert::lm::MemoryMappedFile> model_file)
171178
: memory_mapped_file_(std::move(model_file)) {}
172179

180+
ModelAssets::ModelAssets(
181+
std::shared_ptr<litert::lm::MemoryMappedFile> model_file,
182+
absl::string_view model_path)
183+
: path_(model_path), memory_mapped_file_(std::move(model_file)) {}
184+
173185
absl::StatusOr<absl::string_view> ModelAssets::GetPath() const {
174186
if (!path_.empty()) {
175187
return path_;

runtime/executor/executor_settings_base.h

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,9 @@ class ModelAssets {
111111
static absl::StatusOr<ModelAssets> Create(absl::string_view model_path);
112112
static absl::StatusOr<ModelAssets> Create(
113113
std::shared_ptr<MemoryMappedFile> model_file);
114+
static absl::StatusOr<ModelAssets> Create(
115+
std::shared_ptr<MemoryMappedFile> model_file,
116+
absl::string_view model_path);
114117

115118
// Convenience factory function to create a ModelAssets with both a model
116119
// path and file. Will use the scoped file if both are provided.
@@ -141,6 +144,8 @@ class ModelAssets {
141144
absl::string_view model_path);
142145
explicit ModelAssets(absl::string_view model_path);
143146
explicit ModelAssets(std::shared_ptr<MemoryMappedFile> model_file);
147+
explicit ModelAssets(std::shared_ptr<MemoryMappedFile> model_file,
148+
absl::string_view model_path);
144149

145150
// TODO: b/417814685 - Consider supporting multiple model files if the need
146151
// case arises.

0 commit comments

Comments
 (0)