diff --git a/docs/requirements.txt b/docs/requirements.txt
index 40e7c8e..2fcd959 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -1,3 +1,3 @@
 nats_bench>=1.4
 torchvision
-pytorch/pytorch
+torch