diff --git a/dyana/loaders/automodel/requirements.txt b/dyana/loaders/automodel/requirements.txt index ac5ee45..d0b9bc3 100644 --- a/dyana/loaders/automodel/requirements.txt +++ b/dyana/loaders/automodel/requirements.txt @@ -1,3 +1,3 @@ -transformers==4.57.6 +transformers==5.4.0 torch==2.11.0 accelerate==1.13.0 \ No newline at end of file diff --git a/dyana/loaders/base/dyana-requirements-gpu.txt b/dyana/loaders/base/dyana-requirements-gpu.txt index fd67633..4591ff1 100644 --- a/dyana/loaders/base/dyana-requirements-gpu.txt +++ b/dyana/loaders/base/dyana-requirements-gpu.txt @@ -1,3 +1,3 @@ -psutil==6.1.1 +psutil==7.2.2 torch==2.11.0 pydantic==2.12.5 \ No newline at end of file diff --git a/dyana/loaders/base/dyana-requirements.txt b/dyana/loaders/base/dyana-requirements.txt index 90a9d3e..c44ec0d 100644 --- a/dyana/loaders/base/dyana-requirements.txt +++ b/dyana/loaders/base/dyana-requirements.txt @@ -1,2 +1,2 @@ -psutil==6.1.1 +psutil==7.2.2 pydantic==2.12.5 \ No newline at end of file diff --git a/dyana/loaders/megatron/Dockerfile b/dyana/loaders/megatron/Dockerfile index b2dc99c..551991f 100644 --- a/dyana/loaders/megatron/Dockerfile +++ b/dyana/loaders/megatron/Dockerfile @@ -1,4 +1,4 @@ -FROM nvcr.io/nvidia/pytorch:24.12-py3 +FROM nvcr.io/nvidia/pytorch:26.03-py3 WORKDIR /app