Add config to model info (#135)

huggingface · Jun 22, 2021 · 11d9ab6 · 11d9ab6
1 parent 8e26dc0
commit 11d9ab6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/huggingface_hub/hf_api.py b/src/huggingface_hub/hf_api.py
@@ -84,6 +84,7 @@ def __init__(
         siblings: Optional[
             List[Dict]
         ] = None,  # list of files that constitute the model
+        config: Optional[Dict] = None,  # information about model configuration
         **kwargs,
     ):
         self.modelId = modelId
@@ -94,6 +95,7 @@ def __init__(
         self.siblings = (
             [ModelFile(**x) for x in siblings] if siblings is not None else None
         )
+        self.config = config
         for k, v in kwargs.items():
             setattr(self, k, v)