mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-31 02:02:21 +06:00
[HfApi] Add support for pipeline_tag
This commit is contained in:
parent
0a3d0e02c5
commit
fc24a93e64
@ -80,6 +80,7 @@ class ModelInfo:
|
||||
author: Optional[str] = None,
|
||||
downloads: Optional[int] = None,
|
||||
tags: List[str] = [],
|
||||
pipeline_tag: Optional[str] = None,
|
||||
siblings: Optional[List[Dict]] = None, # list of files that constitute the model
|
||||
**kwargs
|
||||
):
|
||||
@ -88,6 +89,7 @@ class ModelInfo:
|
||||
self.author = author
|
||||
self.downloads = downloads
|
||||
self.tags = tags
|
||||
self.pipeline_tag = pipeline_tag
|
||||
self.siblings = [S3Object(**x) for x in siblings] if siblings is not None else None
|
||||
for k, v in kwargs.items():
|
||||
setattr(self, k, v)
|
||||
|
Loading…
Reference in New Issue
Block a user