diff --git a/examples/llm-text-summarization/requirements.txt b/examples/llm-text-summarization/requirements.txt index 56b7de2..8b13743 100644 --- a/examples/llm-text-summarization/requirements.txt +++ b/examples/llm-text-summarization/requirements.txt @@ -1,4 +1,4 @@ -torch==2.2.0 +torch==2.6.0 transformers datasets peft==0.4.0 diff --git a/examples/llm-text-summarization/training/requirements.txt b/examples/llm-text-summarization/training/requirements.txt index 917307b..6a2d1b7 100644 --- a/examples/llm-text-summarization/training/requirements.txt +++ b/examples/llm-text-summarization/training/requirements.txt @@ -1,5 +1,5 @@ # sagemaker estimators requires training.py and requirement in strict root folder, not -torch==2.2.0 +torch==2.6.0 transformers datasets peft==0.4.0