jupyter notebook fix to upload ldm safetensors instead of ckpt

2023-11-18 13:33:22 -05:00 · 2023-11-18 13:33:22 -05:00 · af74c7f5e1
parent 31e1619a13
commit af74c7f5e1
1 changed files with 1 additions and 1 deletions
--- 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",