upgrade torch to 2.3.1

2024-08-22 19:36:37 -04:00
parent 72a7c4c5c8
commit a2e2cec4c4
2 changed files with 992 additions and 1048 deletions
@@ -42,7 +42,7 @@ if %errorlevel% == 0 (
 	CALL mim install mmengine mmcv>=2.0.1 mmdet>=3.1.0 mmpose>=1.1.0

 	rem install CUDA torch
-	CALL pip install torch==2.0.1+cu118 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
+	CALL pip install torch==2.3.1+cu118 torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu118
 	
    CALL python main.py --disable-smart-memory --preview-method=taesd --force-fp16 --use-pytorch-cross-attention
 ) else (