[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:
Joao Gante
2025-02-20 17:56:11 +00:00
committed by GitHub
parent effaef334b
commit 27d1707586
5 changed files with 13 additions and 7 deletions

View File

@@ -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"]