Remove local scope import of multihost_utils

Author: amepas

src/maxdiffusion/pipelines/flux/flux2klein_pipeline.py

@@ -360,7 +360,6 @@ def step_fn(step_idx, current_latents):
     images_rgb = decoded_out.sample
     images_rgb.block_until_ready()
     if jax.process_count() > 1:
-      from jax.experimental import multihost_utils
       images_rgb = multihost_utils.process_allgather(images_rgb, tiled=True)
 
     trace["vae_decode"] = time.perf_counter() - t0