diff --git a/container-images/cpu/mft-fine-tuning/pytorch/src/requirements.txt b/container-images/cpu/mft-fine-tuning/pytorch/src/requirements.txt index df4d70d4e..00904c90b 100644 --- a/container-images/cpu/mft-fine-tuning/pytorch/src/requirements.txt +++ b/container-images/cpu/mft-fine-tuning/pytorch/src/requirements.txt @@ -9,6 +9,6 @@ psutil==6.1.0 pynvml==11.5.3 rich==14.2.0 thejsonlogger==0.0.3 -torch==2.5.1 +torch==2.8.0 transformers==4.46.1 trl==0.11.4