Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

jupyter notebook fix to upload ldm safetensors instead of ckpt

This commit is contained in:
Victor Hall 2023-11-18 13:33:22 -05:00
parent 31e1619a13
commit af74c7f5e1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Train_JupyterLab.ipynb
View File

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