File tree Expand file tree Collapse file tree
finetuning/src/integrations Expand file tree Collapse file tree Original file line number Diff line number Diff 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
Original file line number Diff line number Diff line change 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 &
Original file line number Diff line number Diff line change @@ -11,5 +11,5 @@ prometheus-fastapi-instrumentator
1111pydantic == 2.9.2
1212pydub
1313shortuuid
14- transformers == 4.48 .0
14+ transformers >= 4.50 .0
1515uvicorn
Original file line number Diff line number Diff line change @@ -8,5 +8,5 @@ opentelemetry-sdk
88prometheus-fastapi-instrumentator
99pydantic==2.9.2
1010shortuuid
11- transformers==4.48 .0
11+ transformers>=4.50 .0
1212uvicorn
Original file line number Diff line number Diff line change @@ -30,7 +30,7 @@ RUN curl -fsSL -v -o miniforge.sh -O https://github.com/conda-forge/miniforge/re
3030RUN git clone https://github.com/jianan-gu/sglang -b llama4_optimzed_cpu_r1
3131RUN . ~/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
3535RUN 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 && \
You can’t perform that action at this time.
0 commit comments