diff --git a/tf_adapter/python/npu_bridge/embedding/embedding_service.py b/tf_adapter/python/npu_bridge/embedding/embedding_service.py index ef03bae87ea1eda89de9131b7d5ad79647bbf4df..712b02f8e5f8b91b12572bcd58d3f67de5408f8b 100644 --- a/tf_adapter/python/npu_bridge/embedding/embedding_service.py +++ b/tf_adapter/python/npu_bridge/embedding/embedding_service.py @@ -1793,7 +1793,8 @@ class ESWorker: gen_npu_cpu_ops.embedding_feature_mapping_import(file_path=path, table_name=x[:-2], feature_size=feature_size, - embedding_dim=self._small_table_embedding_dim) + embedding_dim=self._small_table_embedding_dim, + only_offset_flag=True) feature_mapping_insert = \ gen_npu_cpu_ops.embedding_feature_mapping_insert(table_name=x[:-2], feature_id=feature_id,