From 567fa0a8445b782179c69444a8608155987ba251 Mon Sep 17 00:00:00 2001 From: Oleg Pipikin Date: Tue, 10 Dec 2024 08:57:27 +0000 Subject: [PATCH] Fix for GPU --- src/plugins/intel_gpu/src/plugin/plugin.cpp | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/plugins/intel_gpu/src/plugin/plugin.cpp b/src/plugins/intel_gpu/src/plugin/plugin.cpp index c8839472a6d962..9f9c9692b57b42 100644 --- a/src/plugins/intel_gpu/src/plugin/plugin.cpp +++ b/src/plugins/intel_gpu/src/plugin/plugin.cpp @@ -329,6 +329,12 @@ std::shared_ptr Plugin::import_model(std::istream& model, _orig_config.erase(it); } + std::shared_ptr model_buffer; + if (_orig_config.count(ov::internal::cached_model_buffer.name())) { + model_buffer = _orig_config.at(ov::internal::cached_model_buffer.name()).as>(); + _orig_config.erase(ov::internal::cached_model_buffer.name()); + } + ExecutionConfig config = m_configs_map.at(device_id); config.set_user_property(_orig_config); config.apply_user_properties(context_impl->get_engine().get_device_info());