riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-01 02:31:11 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

[HfApi] Add support for pipeline_tag

Browse Source
This commit is contained in:
Julien Chaumond 2020-06-24 16:54:00 +00:00
parent 0a3d0e02c5
commit fc24a93e64
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/transformers/hf_api.py
View File

@ -80,6 +80,7 @@ class ModelInfo:
author: Optional[str] = None, author: Optional[str] = None,
downloads: Optional[int] = None, downloads: Optional[int] = None,
tags: List[str] = [], tags: List[str] = [],
pipeline_tag: Optional[str] = None,
siblings: Optional[List[Dict]] = None, # list of files that constitute the model siblings: Optional[List[Dict]] = None, # list of files that constitute the model
**kwargs **kwargs
): ):
@ -88,6 +89,7 @@ class ModelInfo:
self.author = author self.author = author
self.downloads = downloads self.downloads = downloads
self.tags = tags self.tags = tags
self.pipeline_tag = pipeline_tag
self.siblings = [S3Object(**x) for x in siblings] if siblings is not None else None self.siblings = [S3Object(**x) for x in siblings] if siblings is not None else None
for k, v in kwargs.items(): for k, v in kwargs.items():
setattr(self, k, v) setattr(self, k, v)