Fix torch and torchvision version compatibility

requirements.txt

@@ -118,8 +118,8 @@ tiktoken==0.8.0
 timeout-decorator==0.5.0
 tokenizers==0.19.1
 tomli==2.2.1
-torch>=2.1.2
-torchaudio>=2.1.2
+torch==2.0.1
+torchaudio==2.0.2
 torchvision==0.15.2
 tqdm==4.67.1
 transformers==4.41.1