From 86b6d4e9d2fc365df04292128d0e79cb7fd74b9e Mon Sep 17 00:00:00 2001
From: Tim Vieira <tim.f.vieira@gmail.com>
Date: Tue, 18 Jun 2024 13:49:53 -0400
Subject: [PATCH] yikes there are even more random seeds!

---
 genparse/steer.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/genparse/steer.py b/genparse/steer.py
index c718a390..006349b9 100644
--- a/genparse/steer.py
+++ b/genparse/steer.py
@@ -29,9 +29,11 @@
 
 def set_seed(seed):
     random.seed(seed)
+    np.random.seed(seed)
     torch.manual_seed(seed)
     transformers.set_seed(seed)
-    np.random.seed(seed)
+    if torch.cuda.is_available():
+        torch.cuda.manual_seed_all(seed)
 
 
 # ____________________________________________________________________________________