Re-enable textgen_test.

BuildKite images are now preloaded with the GPU Docker images, so this
should no longer be an issue.

Startblock:
  cl/607144026 is submitted
PiperOrigin-RevId: 607418077

Makefile

@@ -293,8 +293,7 @@ cos-gpu-smoke-tests: gpu-smoke-images $(RUNTIME_BIN)
 # This is a superset of those needed for smoke tests.
 # It includes non-GPU images that are used as part of GPU tests,
 # e.g. busybox and python.
-# TODO(b/306706609): Add "load-gpu_ollama" and "load-basic_busybox" back here.
-gpu-images: gpu-smoke-images load-gpu_pytorch load-basic_python
+gpu-images: gpu-smoke-images load-gpu_pytorch load-gpu_ollama load-basic_busybox load-basic_python
 .PHONY: gpu-images
 
 gpu-all-tests: gpu-images gpu-smoke-tests $(RUNTIME_BIN)

test/gpu/textgen_test.go

@@ -66,7 +66,6 @@ func runSandboxedPython(ctx context.Context, logger testutil.Logger, code string
 // Then it asks it to write a unit test that verifies that
 // this text is a correct translation.
 func TestLLM(t *testing.T) {
-	t.Skip("Test is disabled until support for large images is better.")
 	ctx := context.Background()
 	// Run the LLM.
 	llmContainer := dockerutil.MakeContainer(ctx, t)