diff --git a/pyproject.toml b/pyproject.toml index ddb68fd..152ccbc 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -20,8 +20,8 @@ dependencies = [ gpu = [ # CUDA / GPU-specific deps. NEVER installed in CI. # Adapt versions / wheels to your stack as needed. - "torch==2.9.1+cu130", - "torchvision==0.24.1+cu130", + "torch==2.10.0", + "torchvision==0.25.0", "xformers>=0.0.28", ] notebooks = [