Make SafeTensorsInfo.parameters a Dict instead of List

huggingface · Oct 2, 2024 · 3e52ebe · 3e52ebe
1 parent ce38f46
commit 3e52ebe
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/huggingface_hub/hf_api.py b/src/huggingface_hub/hf_api.py
@@ -358,7 +358,7 @@ def __post_init__(self):  # hack to make TransformersInfo backward compatible
 
 @dataclass
 class SafeTensorsInfo(dict):
-    parameters: List[Dict[str, int]]
+    parameters: Dict[str, int]
     total: int
 
     def __post_init__(self):  # hack to make SafeTensorsInfo backward compatible