update notebook
This commit is contained in:
parent
41ec1ab161
commit
6ec70b69e1
|
@ -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)"
|
||||
]
|
||||
},
|
||||
|
|
Loading…
Reference in New Issue