From 95ca82d872782a4c3feacf27e2d02169c0c5ad65 Mon Sep 17 00:00:00 2001 From: Etienne Perot Date: Mon, 9 Dec 2024 22:01:01 -0800 Subject: [PATCH] Stable Diffusion benchmark: Specify "latest" tag for container image. Also increase timeout to allow for the time to download the image the first time the benchmark runs. PiperOrigin-RevId: 704545791 --- test/kubernetes/benchmarks/stablediffusion.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/kubernetes/benchmarks/stablediffusion.go b/test/kubernetes/benchmarks/stablediffusion.go index 5bfe239af..a6fdc6ab5 100644 --- a/test/kubernetes/benchmarks/stablediffusion.go +++ b/test/kubernetes/benchmarks/stablediffusion.go @@ -34,7 +34,7 @@ import ( const ( // Container image for Stable Diffusion XL. - stableDiffusionImage = k8s.ImageRepoPrefix + "gpu/stable-diffusion-xl" + stableDiffusionImage = k8s.ImageRepoPrefix + "gpu/stable-diffusion-xl:latest" ) // kubernetesPodRunner implements `stablediffusion.ContainerRunner`. @@ -171,7 +171,7 @@ func RunStableDiffusionXL(ctx context.Context, t *testing.T, k8sCtx k8sctx.Kuber t.Skipf("refiner failed in previous benchmark; skipping benchmark with refiner") } } - testCtx, testCancel := context.WithTimeout(ctx, 15*time.Minute) + testCtx, testCancel := context.WithTimeout(ctx, 50*time.Minute) defer testCancel() prompt := &stablediffusion.XLPrompt{ Query: test.query,