diff --git a/dyana/loaders/automodel/requirements.txt b/dyana/loaders/automodel/requirements.txt index 6e0a7df..8ac6044 100644 --- a/dyana/loaders/automodel/requirements.txt +++ b/dyana/loaders/automodel/requirements.txt @@ -1,3 +1,3 @@ transformers==4.57.6 -torch==2.12.1 +torch==2.13.0 accelerate==1.14.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 43fa2a4..9adca21 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 -torch==2.12.1 +torch==2.13.0 pydantic==2.13.4 \ No newline at end of file diff --git a/dyana/loaders/lora/requirements.txt b/dyana/loaders/lora/requirements.txt index 1cd0192..fd1772a 100644 --- a/dyana/loaders/lora/requirements.txt +++ b/dyana/loaders/lora/requirements.txt @@ -1,2 +1,2 @@ peft==0.19.1 -torch==2.12.1 \ No newline at end of file +torch==2.13.0 \ No newline at end of file diff --git a/dyana/loaders/megatron/requirements.txt b/dyana/loaders/megatron/requirements.txt index fbf1ef1..2a5f1ae 100644 --- a/dyana/loaders/megatron/requirements.txt +++ b/dyana/loaders/megatron/requirements.txt @@ -9,7 +9,7 @@ typing_extensions>=4.0.0 # Megatron DMC dependencies flash-attn==2.8.3.post1 sentencepiece==0.2.1 -hydra-core==1.3.3 +hydra-core==1.3.4 hydra_colorlog==1.2.0 nltk datasets