save tokenizer to each ckpt folder

run_image_caption_flax.py

@@ -999,7 +999,7 @@ def main():
         if jax.process_index() == 0:
             params = jax.device_get(jax.tree_map(lambda x: x[0], state.params))
             model.save_pretrained(os.path.join(training_args.output_dir, f'ckpt_{epoch+1}'), params=params)
-            tokenizer.save_pretrained(training_args.output_dir)
+            tokenizer.save_pretrained(os.path.join(training_args.output_dir, f'ckpt_{epoch+1}'))
             if training_args.push_to_hub:
                 repo.push_to_hub(commit_message=f"Saving weights and logs of epoch {epoch}", blocking=False)