diff --git a/launch.py b/launch.py
index eae5b19e..d7935a73 100644
--- a/launch.py
+++ b/launch.py
@@ -29,7 +29,7 @@ TRY_INSTALL_XFORMERS = False
 def prepare_environment():
     torch_index_url = os.environ.get('TORCH_INDEX_URL', "https://download.pytorch.org/whl/cu121")
     torch_command = os.environ.get('TORCH_COMMAND',
-                                   f"pip install torch==2.1.0 torchvision==0.16.0 --extra-index-url {torch_index_url}")
+                                   f"pip install torch==2.7.1+cu128 orchvision==0.22.1+cu128 --extra-index-url {torch_index_url}")
     requirements_file = os.environ.get('REQS_FILE', "requirements_versions.txt")
 
     print(f"Python {sys.version}")