diff --git a/requirements.txt b/requirements.txt
index decdc195..0e65289d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -18,7 +18,5 @@ dbus-python
 toml
 python-dateutil
 websockets
-torch
-torchvision
-stable_baselines3
+git+https://github.com/openai/baselines
 RPi.GPIO