From 190a8691b10321a961405607ecb1ec3a23540c6f Mon Sep 17 00:00:00 2001
From: Pedro Orlando Acosta Pereira <poacosta87@gmail.com>
Date: Fri, 10 Jan 2025 12:52:34 +0100
Subject: [PATCH] Fixing ValueError when using `save_weights_only=True` in
 `ModelCheckpoint`

---
 site/en/tutorials/audio/music_generation.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/site/en/tutorials/audio/music_generation.ipynb b/site/en/tutorials/audio/music_generation.ipynb
index e1423ef7cf2..0667c761f73 100644
--- a/site/en/tutorials/audio/music_generation.ipynb
+++ b/site/en/tutorials/audio/music_generation.ipynb
@@ -999,7 +999,7 @@
       "source": [
         "callbacks = [\n",
         "    tf.keras.callbacks.ModelCheckpoint(\n",
-        "        filepath='./training_checkpoints/ckpt_{epoch}',\n",
+        "        filepath='./training_checkpoints/ckpt_{epoch}.weights.h5',\n",
         "        save_weights_only=True),\n",
         "    tf.keras.callbacks.EarlyStopping(\n",
         "        monitor='loss',\n",