diff --git a/requirements.dev.txt b/requirements.dev.txt index 976ab2bb..87fcfecc 100644 --- a/requirements.dev.txt +++ b/requirements.dev.txt @@ -1,4 +1,4 @@ -pytest==8.3.1 +pytest==8.3.2 mypy==1.11.0 black==24.4.2 ruff==0.5.4 diff --git a/requirements.rocm.txt b/requirements.rocm.txt index bc1b4408..6272e55f 100644 --- a/requirements.rocm.txt +++ b/requirements.rocm.txt @@ -1,5 +1,5 @@ -numpy==1.26.4 -torch==2.3.1+rocm6.0 +numpy==2.0.1 +torch==2.4.0 horde_sdk~=0.14.0 horde_safety~=0.2.3 diff --git a/requirements.txt b/requirements.txt index edc910a6..e34c1dae 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,8 +3,8 @@ horde_sdk>=0.14.0 horde_model_reference>=0.8.1 horde_safety~=0.2.3 -numpy==1.26.4 -torch==2.4.0+cu124 +numpy==2.0.1 +torch==2.4.0 mkl_fft