diff --git a/pyproject.toml b/pyproject.toml index 340b226..300666b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -33,7 +33,7 @@ finetune = [ "imagesize==1.4.1", "ms-swift==3.5.3", "qwen_vl_utils>=0.0.6", - "torch==2.6.0", + "torch==2.8.0", "vllm==0.8.5.post1" ] comfor = ["timm>=0.9.12", "torchvision>=0.17.1"]