From d2a7ffb5fe3153871435e112ad658f5b5bf77031 Mon Sep 17 00:00:00 2001
From: mscs <markscsmith@gmail.com>
Date: Wed, 17 Apr 2024 20:55:06 -0400
Subject: [PATCH] =?UTF-8?q?Eureka!=20Turns=20out=20rocm=20driver=20version?=
 =?UTF-8?q?s=20weren't=20the=20issue.=20sb3=20above=20this=20version=20now?=
 =?UTF-8?q?=20attempts=20to=20use=20torch.load=20weights=5Fonly=3DTrue=20r?=
 =?UTF-8?q?elated=20to=20https://github.com/DLR-RM/stable-baselines3/issue?=
 =?UTF-8?q?s/1852=20=F0=9F=98=AE=E2=80=8D=F0=9F=92=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 requirements.txt | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 91a069fd..b7cea972 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,14 +5,13 @@ numba
 numpy
 pandas
 Pillow
-pyboy==2.0.0
+pyboy>=2.0.0
 scipy
 Shimmy
-stable-baselines3
+stable-baselines3[extra]==2.2.1
 tensorboard
 tensorflow
 timg
 torch
-torchaudio
 torchvision
 tqdm