diff --git a/Train_JupyterLab.ipynb b/Train_JupyterLab.ipynb
index 1163916..5857949 100644
--- a/Train_JupyterLab.ipynb
+++ b/Train_JupyterLab.ipynb
@@ -250,7 +250,7 @@
     "from huggingface_hub import HfApi\n",
     "from ipywidgets import *\n",
     "\n",
-    "all_ckpts = [f for f in glob.glob(\"*.ckpt\")]\n",
+    "all_ckpts = [f for f in glob.glob(\"*.safetensors\")]\n",
     "  \n",
     "ckpt_picker = SelectMultiple(options=all_ckpts, layout=Layout(width=\"600px\")) \n",
     "hfuser = Text(placeholder='Your HF user name')\n",