diff --git a/maxdiffusion_jax_stable_stack_tpu.Dockerfile b/maxdiffusion_jax_stable_stack_tpu.Dockerfile index 5d94695a..6d7c35fe 100644 --- a/maxdiffusion_jax_stable_stack_tpu.Dockerfile +++ b/maxdiffusion_jax_stable_stack_tpu.Dockerfile @@ -1,7 +1,7 @@ ARG JAX_STABLE_STACK_BASEIMAGE # JAX Stable Stack Base Image -From $JAX_STABLE_STACK_BASEIMAGE +FROM $JAX_STABLE_STACK_BASEIMAGE ARG COMMIT_HASH