Fix max_utils import in generate_flux2klein.py

Author: amepas

src/maxdiffusion/generate_flux2klein.py

@@ -31,6 +31,7 @@
 
 from maxdiffusion import pyconfig
 from maxdiffusion import max_logging
+from maxdiffusion import max_utils
 from maxdiffusion.max_utils import create_device_mesh
 from maxdiffusion.train_utils import transformer_engine_context