From 80424839f8fa70e844397fc0064850d8c94ea38e Mon Sep 17 00:00:00 2001
From: Aditya Borikar <adityaborikar2@gmail.com>
Date: Thu, 3 Oct 2024 01:49:19 -0700
Subject: [PATCH] Make SafeTensorsInfo.parameters a Dict instead of List
 (#2585)

---
 src/huggingface_hub/hf_api.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/huggingface_hub/hf_api.py b/src/huggingface_hub/hf_api.py
index a74bdef0eb..ed97cc1063 100644
--- 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