diff --git a/requirements.txt b/requirements.txt
index 0690bef6..0ac50640 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -5,7 +5,7 @@ memory-profiler
 bitsandbytes==0.41.3.post2
 typing==3.7.4.3
 einops-exts==0.0.4
-torchvision==0.16.1
+torchvision
 tokenmonster==1.1.12
 accelerate
 datasets==2.10.1