update notebook

This commit is contained in:
Victor Hall 2022-11-23 21:46:47 -05:00
parent 41ec1ab161
commit 6ec70b69e1
1 changed files with 2 additions and 1 deletions

View File

@ -143,6 +143,7 @@
"metadata": {},
"outputs": [],
"source": [
"# Default is 1.5 with the new VAE, you can change this to another file on huggingface if you want:\n",
"from huggingface_hub import hf_hub_download\n",
"downloaded_model_path = hf_hub_download(\n",
" repo_id=\"panopstor/EveryDream\",\n",
@ -234,7 +235,7 @@
"outputs": [],
"source": [
"# run the trainer, wait until it finishes then SCROLL DOWN to the next cell\n",
"!python main.py --base {my_yaml} -t --actual_resume {downloaded_model_path} -n test --data_root input\n",
"!python main.py --base {my_yaml} -t --actual_resume whatever.ckpt -n test --data_root input\n",
"print(finish_msg)"
]
},