diff --git a/pyproject.toml b/pyproject.toml index d5c8c03163..5394d31354 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -70,7 +70,7 @@ dependencies = [ "taming-transformers-rom1504", "test-tube>=0.7.5", "torch-fidelity", - "torch>=1.13.1", + "torch~=1.13.1", "torchmetrics", "torchvision>=0.14.1", "transformers~=4.26",