From 3577bf2688463afc7b2a2eba6e5cb62640fa6f63 Mon Sep 17 00:00:00 2001
From: Michael Lucas <michael@desktopninjas.com>
Date: Sat, 6 Apr 2024 18:06:14 -0500
Subject: [PATCH] added jaxlib specific version

In some cases, jaxlib will not install exact compatible version number.  specifying version in requirements avoids startup error
---
 requirements.txt | 1 +
 1 file changed, 1 insertion(+)

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