diff --git a/composite_demo/requirements.txt b/composite_demo/requirements.txt
index 0cf6992..56c8484 100644
--- a/composite_demo/requirements.txt
+++ b/composite_demo/requirements.txt
@@ -22,4 +22,4 @@ streamlit>=1.35.0
 tiktoken>=0.7.0
 transformers==4.40.0
 zhipuai>=2.1.0
-
+torchvision>=0.18.1