diff --git a/installer/lib/installer.py b/installer/lib/installer.py index 42079cd90c..96d8ce55e3 100644 --- a/installer/lib/installer.py +++ b/installer/lib/installer.py @@ -247,8 +247,8 @@ class InvokeAiInstance: pip[ "install", "--require-virtualenv", - "torch", - "torchvision", + "torch~=2.0.0", + "torchvision>=0.14.1", "--force-reinstall", "--find-links" if find_links is not None else None, find_links, diff --git a/pyproject.toml b/pyproject.toml index 0ab1e0ede6..fd671fee23 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -38,9 +38,9 @@ dependencies = [ "albumentations", "click", "clip_anytorch", # replacing "clip @ https://github.com/openai/CLIP/archive/eaa22acb90a5876642d0507623e859909230a52d.zip", - "compel==1.0.5", + "compel~=1.1.5", "datasets", - "diffusers[torch]==0.15.1", + "diffusers[torch]~=0.16.1", "dnspython==2.2.1", "einops", "eventlet", @@ -72,7 +72,7 @@ dependencies = [ "scikit-image>=0.19", "send2trash", "test-tube>=0.7.5", - "torch>=1.13.1", + "torch~=2.0.0", "torchvision>=0.14.1", "torchmetrics", "transformers~=4.26", @@ -92,7 +92,7 @@ dependencies = [ ] "test" = ["pytest>6.0.0", "pytest-cov"] "xformers" = [ - "xformers~=0.0.16; sys_platform!='darwin'", + "xformers~=0.0.19; sys_platform!='darwin'", "triton; sys_platform=='linux'", ]