File tree Expand file tree Collapse file tree 2 files changed +17
-0
lines changed
Expand file tree Collapse file tree 2 files changed +17
-0
lines changed Original file line number Diff line number Diff 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+
162169ModelAssets::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+
173185absl::StatusOr<absl::string_view> ModelAssets::GetPath () const {
174186 if (!path_.empty ()) {
175187 return path_;
Original file line number Diff line number Diff 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.
You can’t perform that action at this time.
0 commit comments