fix path for saving image.

Author: jfacevedo-google

examples/research_projects/pytorch_xla/inference/flux/generate_flux.py

@@ -79,7 +79,7 @@ def _main(index, args, text_pipe, ckpt_id):
             logger.info(f"inference time: {inference_time}")
         times.append(inference_time)
     logger.info(f'avg. inference over {args.itters} iterations took {sum(times)/len(times)} sec.')
-    image.save(f'/home/tmp/inference_out-{index}.png')
+    image.save(f'/tmp/inference_out-{index}.png')
     if index == 0:
         metrics_report = met.metrics_report()
         with open(metrics_filepath, 'w+') as fout: