From 823716030728d2271441ea60d4c1e3af2e94b23e Mon Sep 17 00:00:00 2001 From: Andrew Powers-Holmes Date: Mon, 26 Feb 2024 16:49:23 +1100 Subject: [PATCH] fine --- docker-bake.hcl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docker-bake.hcl b/docker-bake.hcl index e363b90..2bc5dda 100644 --- a/docker-bake.hcl +++ b/docker-bake.hcl @@ -64,7 +64,7 @@ function torchName { # torch version to torch name function torchSpec { params = [version] - result = regex_replace(version, "^(\\d+)\\.(\\d+)\\.(\\d+).*", "torch~=$1.$2.$3") + result = regex_replace(version, "^(\\d+)\\.(\\d+)\\.(\\d+).*", "torch==$1.$2.$3") } # build a tag for an image from this repo @@ -130,7 +130,7 @@ target "base" { matrix = { torch = [ { - version = "2.1.0" + version = "2.1.1" index = "https://download.pytorch.org/whl" xformers = "xformers>=0.0.22" },