updated requirements for torch

hkproj · hkproj · commit 788cdeda3297 · 2023-12-08T11:16:48.000+08:00
diff --git a/requirements.txt b/requirements.txt
@@ -1,6 +1,8 @@
 ## Use python 3.9
 
 torch==2.0.1
+torchvision==0.15.2 
+torchaudio==2.0.2
 torchtext==0.15.2
 datasets==2.15.0
 tokenizers==0.13.3
