Add PT + TF automatic builds (#15860)
* Add PT + TF automatic builds * Apply suggestions from code review Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com> * Wrap up Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>
This commit is contained in:
30
setup.py
30
setup.py
@@ -309,6 +309,36 @@ extras["docs_specific"] = []
|
||||
# "docs" needs "all" to resolve all the references
|
||||
extras["docs"] = extras["all"] + extras["docs_specific"]
|
||||
|
||||
extras["dev-torch"] = (
|
||||
extras['testing']
|
||||
+ extras['torch']
|
||||
+ extras["sentencepiece"]
|
||||
+ extras["tokenizers"]
|
||||
+ extras["torch-speech"]
|
||||
+ extras["vision"]
|
||||
+ extras["integrations"]
|
||||
+ extras["timm"]
|
||||
+ extras["codecarbon"]
|
||||
+ extras["quality"]
|
||||
+ extras["ja"]
|
||||
+ extras["docs_specific"]
|
||||
+ extras["sklearn"]
|
||||
+ extras["modelcreation"]
|
||||
+ extras["onnxruntime"]
|
||||
)
|
||||
extras["dev-tensorflow"] = (
|
||||
extras['testing']
|
||||
+ extras['tf']
|
||||
+ extras["sentencepiece"]
|
||||
+ extras["tokenizers"]
|
||||
+ extras["vision"]
|
||||
+ extras["quality"]
|
||||
+ extras["docs_specific"]
|
||||
+ extras["sklearn"]
|
||||
+ extras["modelcreation"]
|
||||
+ extras["onnx"]
|
||||
+ extras["tf-speech"]
|
||||
)
|
||||
extras["dev"] = (
|
||||
extras["all"]
|
||||
+ extras["testing"]
|
||||
|
||||
Reference in New Issue
Block a user