diff --git a/docker/tf-light.dockerfile b/docker/tf-light.dockerfile index 67dc928c22..e58db2aa00 100644 --- a/docker/tf-light.dockerfile +++ b/docker/tf-light.dockerfile @@ -7,6 +7,6 @@ RUN apt-get install -y cmake ENV UV_PYTHON=/usr/local/bin/python RUN pip --no-cache-dir install uv && uv venv && uv pip install --no-cache-dir -U pip setuptools RUN pip install --upgrade --no-cache-dir "git+https://github.com/huggingface/transformers.git@${REF}#egg=transformers[tf-cpu,sklearn,testing,sentencepiece,tf-speech,vision]" -RUN uv pip install --no-cache-dir "protobuf==3.20.3" +RUN uv pip install --no-cache-dir "protobuf==3.20.3" RUN pip uninstall -y transformers -RUN apt-get clean && rm -rf /var/lib/apt/lists/* && apt-get autoremove && apt-get autoclean \ No newline at end of file +RUN apt-get clean && rm -rf /var/lib/apt/lists/* && apt-get autoremove && apt-get autoclean diff --git a/docker/torch-light.dockerfile b/docker/torch-light.dockerfile index c1a8f8ac0f..5e71afe978 100644 --- a/docker/torch-light.dockerfile +++ b/docker/torch-light.dockerfile @@ -7,5 +7,5 @@ ENV UV_PYTHON=/usr/local/bin/python RUN pip --no-cache-dir install uv && uv venv && uv pip install --no-cache-dir -U pip setuptools RUN pip install --no-cache-dir 'torch' 'torchvision' 'torchaudio' --index-url https://download.pytorch.org/whl/cpu RUN uv pip install --no-deps timm accelerate --extra-index-url https://download.pytorch.org/whl/cpu -RUN uv pip install --no-cache-dir librosa "git+https://github.com/huggingface/transformers.git@${REF}#egg=transformers[sklearn,sentencepiece,vision,testing,tiktoken]" -RUN pip uninstall -y transformers \ No newline at end of file +RUN uv pip install --no-cache-dir librosa "git+https://github.com/huggingface/transformers.git@${REF}#egg=transformers[sklearn,sentencepiece,vision,testing,tiktoken,num2words]" +RUN pip uninstall -y transformers diff --git a/docs/source/en/_toctree.yml b/docs/source/en/_toctree.yml index 4b34ccf0e3..06a6f172fa 100644 --- a/docs/source/en/_toctree.yml +++ b/docs/source/en/_toctree.yml @@ -965,6 +965,8 @@ title: Segment Anything - local: model_doc/siglip title: SigLIP + - local: model_doc/smolvlm + title: SmolVLM - local: model_doc/speech-encoder-decoder title: Speech Encoder Decoder Models - local: model_doc/tapas diff --git a/setup.py b/setup.py index 9c207fb816..e44c6703b4 100644 --- a/setup.py +++ b/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"] diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 3a35919e97..61b322b736 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -36,8 +36,9 @@ deps = { "keras": "keras>2.9,<2.16", "keras-nlp": "keras-nlp>=0.3.1,<0.14.0", "librosa": "librosa", - "nltk": "nltk<=3.8.1", "natten": "natten>=0.14.6,<0.15.0", + "nltk": "nltk<=3.8.1", + "num2words": "num2words", "numpy": "numpy>=1.17", "onnxconverter-common": "onnxconverter-common", "onnxruntime-tools": "onnxruntime-tools>=1.4.2",