From 1b267eb70cd8dcb1f3d7c924a9b7cb19779deb04 Mon Sep 17 00:00:00 2001 From: Rajat Sen Date: Mon, 3 Jun 2024 22:32:22 +0000 Subject: [PATCH] minor changes to environment files --- experiments/environment.yml | 2 +- experiments/environment_cpu.yml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/experiments/environment.yml b/experiments/environment.yml index 9e731a4..ea7e746 100644 --- a/experiments/environment.yml +++ b/experiments/environment.yml @@ -20,7 +20,7 @@ dependencies: - git+https://github.com/amazon-science/chronos-forecasting.git - praxis - paxml - - jax[cuda12] + - jax[cuda12]==0.4.26 - einshape - python-dotenv - nixtla>=0.5.1 diff --git a/experiments/environment_cpu.yml b/experiments/environment_cpu.yml index 7cc3912..7d5a0f6 100644 --- a/experiments/environment_cpu.yml +++ b/experiments/environment_cpu.yml @@ -20,7 +20,7 @@ dependencies: - git+https://github.com/amazon-science/chronos-forecasting.git - praxis - paxml - - jax[cpu] + - jax[cpu]==0.4.26 - einshape - python-dotenv - nixtla>=0.5.1