Skip to content

Commit fd7b6b7

Browse files
authored
Upgrade Transformers to minimum v4.50.0 everywhere (#1748)
* Upgrade Transformers to minimum v4.50.0 everywhere Signed-off-by: Abolfazl Shahbazi <12436063+ashahba@users.noreply.github.com> * use minimum version 4.50.0 for transformers Signed-off-by: Abolfazl Shahbazi <12436063+ashahba@users.noreply.github.com> --------- Signed-off-by: Abolfazl Shahbazi <12436063+ashahba@users.noreply.github.com>
1 parent 424da96 commit fd7b6b7

5 files changed

Lines changed: 5 additions & 5 deletions

File tree

comps/finetuning/src/integrations/llm_on_ray/finetune/finetune.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -560,7 +560,7 @@ def main(external_config=None):
560560
}
561561

562562
if config["General"]["gpt_base_model"] is True:
563-
runtime_env["pip"] = ["transformers==4.26.0"]
563+
runtime_env["pip"] = ["transformers>=4.50.0"]
564564

565565
if device == "gpu":
566566
num_cpus = resources_per_worker["CPU"] * num_training_workers + 1 # additional 1 for head worker

comps/finetuning/src/integrations/xtune/prepare_xtune.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ else
3939
pip install matplotlib
4040
pip install -e ".[metrics]"
4141
pip install --no-cache-dir --force-reinstall intel-extension-for-pytorch==2.6.10+xpu oneccl_bind_pt==2.6.0+xpu --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/xpu/us/
42-
pip install "transformers<=4.49.0" optimum "auto_gptq>=0.5.0"
42+
pip install "transformers>=4.50.0" optimum "auto_gptq>=0.5.0"
4343
echo "start llamafactory webui"
4444
if [ -z $GUI ]; then
4545
ZE_AFFINITY_MASK=0 llamafactory-cli webui &

comps/third_parties/llama-vision/src/requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,5 +11,5 @@ prometheus-fastapi-instrumentator
1111
pydantic==2.9.2
1212
pydub
1313
shortuuid
14-
transformers==4.48.0
14+
transformers>=4.50.0
1515
uvicorn

comps/third_parties/llama-vision/src/requirements_tp.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,5 +8,5 @@ opentelemetry-sdk
88
prometheus-fastapi-instrumentator
99
pydantic==2.9.2
1010
shortuuid
11-
transformers==4.48.0
11+
transformers>=4.50.0
1212
uvicorn

comps/third_parties/sglang/src/Dockerfile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ RUN curl -fsSL -v -o miniforge.sh -O https://github.com/conda-forge/miniforge/re
3030
RUN git clone https://github.com/jianan-gu/sglang -b llama4_optimzed_cpu_r1
3131
RUN . ~/miniforge3/bin/activate && conda create -n sglang python=3.10 && conda activate sglang && \
3232
cd sglang && pip install -e "python[all_cpu]" && cd .. && conda install -y libsqlite=3.48.0 && \
33-
pip uninstall -y triton && pip uninstall -y transformers && pip install transformers==4.51.1 && \
33+
pip uninstall -y triton && pip uninstall -y transformers && pip install transformers>=4.50.0 && \
3434
pip install triton==3.1 && pip install intel-openmp==2024.2.0 && pip install transformers
3535
RUN git clone https://github.com/vllm-project/vllm.git -b v0.6.4.post1 && cd vllm && apt-get install -y libnuma-dev && \
3636
. ~/miniforge3/bin/activate && conda activate sglang && \

0 commit comments

Comments
 (0)