Skip to content

Commit 11d9ab6

Browse files
authored
Add config to model info (#135)
1 parent 8e26dc0 commit 11d9ab6

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/huggingface_hub/hf_api.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,7 @@ def __init__(
8484
siblings: Optional[
8585
List[Dict]
8686
] = None, # list of files that constitute the model
87+
config: Optional[Dict] = None, # information about model configuration
8788
**kwargs,
8889
):
8990
self.modelId = modelId
@@ -94,6 +95,7 @@ def __init__(
9495
self.siblings = (
9596
[ModelFile(**x) for x in siblings] if siblings is not None else None
9697
)
98+
self.config = config
9799
for k, v in kwargs.items():
98100
setattr(self, k, v)
99101

0 commit comments

Comments
 (0)