diff --git a/environment-cpu.yml b/environment-cpu.yml index 2419bf35..7dfe823e 100644 --- a/environment-cpu.yml +++ b/environment-cpu.yml @@ -10,6 +10,7 @@ dependencies: - pytorch>=1.8.1 - torchaudio>=0.8 - tqdm>=4.36 + - numpy<2.0 - pip - pip: - diffq>=0.2 diff --git a/environment-cuda.yml b/environment-cuda.yml index 0d61d33d..0b3940d6 100644 --- a/environment-cuda.yml +++ b/environment-cuda.yml @@ -12,6 +12,7 @@ dependencies: - cudatoolkit>=10 - tqdm>=4.36 - pip + - numpy<2.0 - pip: - diffq>=0.2 - dora-search