Add tpc_name to model metadata

sony · Jan 21, 2025 · fe7715a · fe7715a
1 parent d73b4a7
commit fe7715a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/model_compression_toolkit/metadata.py b/model_compression_toolkit/metadata.py
@@ -57,6 +57,7 @@ class TPCVersions:
         tpc_minor_version: str = f'{fqc.tpc.tpc_minor_version}'
         tpc_patch_version: str = f'{fqc.tpc.tpc_patch_version}'
         tpc_platform_type: str = f'{fqc.tpc.tpc_platform_type}'
+        tpc_name: str = f'{fqc.tpc.name}'
         tpc_schema: str = f'{fqc.tpc.SCHEMA_VERSION}'
 
     return asdict(TPCVersions(mct_version))