diff --git a/dyana/loaders/automodel/requirements.txt b/dyana/loaders/automodel/requirements.txt index 74270ac..dca003d 100644 --- a/dyana/loaders/automodel/requirements.txt +++ b/dyana/loaders/automodel/requirements.txt @@ -1,3 +1,3 @@ -transformers==5.18.0 +transformers==5.19.0 torch==2.14.1 accelerate==1.15.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 f3c2a37..04e5ba6 100644 --- a/dyana/loaders/base/dyana-requirements-gpu.txt +++ b/dyana/loaders/base/dyana-requirements-gpu.txt @@ -1,3 +1,3 @@ psutil==7.2.2 torch==2.14.1 -pydantic==2.13.5 \ No newline at end of file +pydantic==2.14.1 \ No newline at end of file diff --git a/dyana/loaders/base/dyana-requirements.txt b/dyana/loaders/base/dyana-requirements.txt index b0e69ea..b7cd846 100644 --- a/dyana/loaders/base/dyana-requirements.txt +++ b/dyana/loaders/base/dyana-requirements.txt @@ -1,2 +1,2 @@ psutil==7.2.2 -pydantic==2.13.5 \ No newline at end of file +pydantic==2.14.1 \ No newline at end of file diff --git a/dyana/loaders/megatron/requirements.txt b/dyana/loaders/megatron/requirements.txt index ebaf0d2..d10f7aa 100644 --- a/dyana/loaders/megatron/requirements.txt +++ b/dyana/loaders/megatron/requirements.txt @@ -7,7 +7,7 @@ packaging>=20.0 typing_extensions>=4.0.0 # Megatron DMC dependencies -flash-attn==2.8.3.post1 +flash-attn==2.8.3.post2 sentencepiece==0.2.2 hydra-core==1.3.7 hydra_colorlog==1.2.0