diff --git a/test/torch_requirements.txt b/test/torch_requirements.txt
index 36fcdd5..74c366b 100644
--- a/test/torch_requirements.txt
+++ b/test/torch_requirements.txt
@@ -1,3 +1,3 @@
-torch==2.1.0
-torchaudio==2.1.0
+torch
+torchaudio
 torchdata