diff --git a/dyana/loaders/automodel/requirements.txt b/dyana/loaders/automodel/requirements.txt index dff4292..6e0a7df 100644 --- a/dyana/loaders/automodel/requirements.txt +++ b/dyana/loaders/automodel/requirements.txt @@ -1,3 +1,3 @@ transformers==4.57.6 -torch==2.12.0 -accelerate==1.13.0 \ No newline at end of file +torch==2.12.1 +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 f6908b6..43fa2a4 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.0 +torch==2.12.1 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 78286ec..1cd0192 100644 --- a/dyana/loaders/lora/requirements.txt +++ b/dyana/loaders/lora/requirements.txt @@ -1,2 +1,2 @@ peft==0.19.1 -torch==2.12.0 \ No newline at end of file +torch==2.12.1 \ No newline at end of file diff --git a/dyana/loaders/megatron/requirements.txt b/dyana/loaders/megatron/requirements.txt index 5852fb4..fbf1ef1 100644 --- a/dyana/loaders/megatron/requirements.txt +++ b/dyana/loaders/megatron/requirements.txt @@ -7,9 +7,9 @@ packaging>=20.0 typing_extensions>=4.0.0 # Megatron DMC dependencies -flash-attn==2.8.3 +flash-attn==2.8.3.post1 sentencepiece==0.2.1 -hydra-core==1.3.2 +hydra-core==1.3.3 hydra_colorlog==1.2.0 nltk datasets