Use yaml to create metadata (#12185)

* Use yaml to create metadata

* Fix typo

* Remove pin
This commit is contained in:
Sylvain Gugger
2021-06-16 13:17:45 -04:00
committed by GitHub
parent 15ef0dc5c6
commit 255a17a089
3 changed files with 52 additions and 33 deletions

View File

@@ -117,6 +117,7 @@ _deps = [
"parameterized",
"protobuf",
"psutil",
"pyyaml",
"pydantic",
"pytest",
"pytest-sugar",
@@ -321,6 +322,7 @@ install_requires = [
deps["huggingface-hub"],
deps["numpy"],
deps["packaging"], # utilities from PyPA to e.g., compare versions
deps["pyyaml"], # used for the model cards metadata
deps["regex"], # for OpenAI GPT
deps["requests"], # for downloading models over HTTPS
deps["sacremoses"], # for XLM