From 7050ed204b8206bb8645c7b7bbef7252f79561b0 Mon Sep 17 00:00:00 2001
From: Eddy <boscolo.eddy@gmail.com>
Date: Tue, 19 Mar 2024 16:48:22 +0100
Subject: [PATCH] Corrected name of package "cuda12-pip" (#194)

The `cuda12-pip` package was wrongly named `cuda12_pip`
in requirements.txt
---
 requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index f6d124e..a612687 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
 dm_haiku==0.0.12
-jax[cuda12_pip]==0.4.25 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
+jax[cuda12-pip]==0.4.25 -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
 numpy==1.26.4
 sentencepiece==0.2.0