diff --git a/src/huggingface_hub/hf_api.py b/src/huggingface_hub/hf_api.py index 2d8d8ddf5f..8cc3ed07d7 100644 --- 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)