netbird/gvisor
0
0
Fork 0
mirror of https://github.com/netbirdio/gvisor.git synced 2026-05-22 17:12:49 -07:00
Code Issues Packages Projects Releases Wiki Activity

Stable Diffusion benchmark: Specify "latest" tag for container image.

Browse Source 
Also increase timeout to allow for the time to download the image the
first time the benchmark runs.

PiperOrigin-RevId: 704545791
This commit is contained in:
Etienne Perot
2024-12-09 22:05:02 -08:00
committed by gVisor bot
parent a55b3b2d90
commit 95ca82d872
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/kubernetes/benchmarks/stablediffusion.go
+2 -2
View File
@@ -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,