From af74c7f5e1e85afadd7c9950a283e012cbb55aa5 Mon Sep 17 00:00:00 2001 From: Victor Hall Date: Sat, 18 Nov 2023 13:33:22 -0500 Subject: [PATCH] jupyter notebook fix to upload ldm safetensors instead of ckpt --- Train_JupyterLab.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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",