Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
transformers
Commits
fc24a93e
Commit
fc24a93e
authored
Jun 24, 2020
by
Julien Chaumond
Browse files
[HfApi] Add support for pipeline_tag
parent
0a3d0e02
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
0 deletions
+2
-0
src/transformers/hf_api.py
src/transformers/hf_api.py
+2
-0
No files found.
src/transformers/hf_api.py
View file @
fc24a93e
...
@@ -80,6 +80,7 @@ class ModelInfo:
...
@@ -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:
...
@@ -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
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment