diff --git a/LLavaTagger/requirements.txt b/LLavaTagger/requirements.txt
index ada4970..c4e265c 100644
--- a/LLavaTagger/requirements.txt
+++ b/LLavaTagger/requirements.txt
@@ -5,7 +5,7 @@ ninja==1.11.1.1
 safetensors==0.4.2
 tokenizers==0.15.2
 transformers
-pytorch
+torch
 opencv-python
 numpy
 tqdm