Eureka! Turns out rocm driver versions weren't the issue. sb3 above t…

…his version now attempts to use torch.load weights_only=True related to DLR-RM/stable-baselines3#1852 😮‍💨
markscsmith · Apr 18, 2024 · d2a7ffb · d2a7ffb
1 parent f2e5c35
commit d2a7ffb
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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