[smolvlm] make CI green (#36306)
* add smolvlm to toctree * add requirements * dev-ci * no docker changes * dev-ci * update torch-light.dockerfile * derp * dev-ci
This commit is contained in:
7
setup.py
7
setup.py
@@ -130,8 +130,9 @@ _deps = [
|
||||
"keras>2.9,<2.16",
|
||||
"keras-nlp>=0.3.1,<0.14.0", # keras-nlp 0.14 doesn't support keras 2, see pin on keras.
|
||||
"librosa",
|
||||
"nltk<=3.8.1",
|
||||
"natten>=0.14.6,<0.15.0",
|
||||
"nltk<=3.8.1",
|
||||
"num2words",
|
||||
"numpy>=1.17",
|
||||
"onnxconverter-common",
|
||||
"onnxruntime-tools>=1.4.2",
|
||||
@@ -314,7 +315,7 @@ extras["torch-vision"] = deps_list("torchvision") + extras["vision"]
|
||||
extras["natten"] = deps_list("natten")
|
||||
extras["codecarbon"] = deps_list("codecarbon")
|
||||
extras["video"] = deps_list("av")
|
||||
|
||||
extras["num2words"] = deps_list("num2words")
|
||||
extras["sentencepiece"] = deps_list("sentencepiece", "protobuf")
|
||||
extras["tiktoken"] = deps_list("tiktoken", "blobfile")
|
||||
extras["testing"] = (
|
||||
@@ -364,6 +365,7 @@ extras["all"] = (
|
||||
+ extras["codecarbon"]
|
||||
+ extras["accelerate"]
|
||||
+ extras["video"]
|
||||
+ extras["num2words"]
|
||||
)
|
||||
|
||||
|
||||
@@ -383,6 +385,7 @@ extras["dev-torch"] = (
|
||||
+ extras["sklearn"]
|
||||
+ extras["modelcreation"]
|
||||
+ extras["onnxruntime"]
|
||||
+ extras["num2words"]
|
||||
)
|
||||
extras["dev-tensorflow"] = (
|
||||
extras["testing"]
|
||||
|
||||
Reference in New Issue
Block a user