tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

simplified colab setup

Browse Source
This commit is contained in:
Brett Kuprel 2022-06-29 16:32:51 -04:00
parent b7b6df23f7
commit 5f931f6899
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
min_dalle.ipynb vendored
View File

@ -37,13 +37,15 @@
},
"outputs": [],
"source": [
"! git clone https://github.com/kuprel/min-dalle\n",
"! mkdir -p /content/min-dalle/pretrained/vqgan/\n",
"! curl https://huggingface.co/dalle-mini/vqgan_imagenet_f16_16384/resolve/main/flax_model.msgpack -L --output /content/min-dalle/pretrained/vqgan/flax_model.msgpack\n",
"! pip install torch flax==0.4.2 wandb\n",
"! wandb login --anonymously\n",
"! wandb artifact get --root=/content/min-dalle/pretrained/dalle_bart_mini dalle-mini/dalle-mini/mini-1:v0\n",
"! wandb artifact get --root=/content/min-dalle/pretrained/dalle_bart_mega dalle-mini/dalle-mini/mega-1-fp16:v14\n"
"%%shell\n",
"\n",
"git clone https://github.com/kuprel/min-dalle\n",
"mkdir -p /content/min-dalle/pretrained/vqgan/\n",
"curl https://huggingface.co/dalle-mini/vqgan_imagenet_f16_16384/resolve/main/flax_model.msgpack -L --output /content/min-dalle/pretrained/vqgan/flax_model.msgpack\n",
"pip install torch flax==0.4.2 wandb\n",
"wandb login --anonymously\n",
"wandb artifact get --root=/content/min-dalle/pretrained/dalle_bart_mini dalle-mini/dalle-mini/mini-1:v0\n",
"wandb artifact get --root=/content/min-dalle/pretrained/dalle_bart_mega dalle-mini/dalle-mini/mega-1-fp16:v14\n"
]
},
{
@ -76,7 +78,7 @@
"base_uri": "https://localhost:8080/"
}
},
"execution_count": 2,
"execution_count": null,
"outputs": [
{
"output_type": "stream",
@ -101,7 +103,7 @@
},
{
"cell_type": "code",
"execution_count": 5,
"execution_count": null,
"metadata": {
"colab": {
"base_uri": "https://localhost:8080/",
@ -169,7 +171,7 @@
"name": "min-dalle",
"provenance": [],
"machine_shape": "hm",
"authorship_tag": "ABX9TyMD0M+1bhwaISRugJcCWvwm",
"authorship_tag": "ABX9TyM7nP04EHVNWqYy09gjYVDl",
"include_colab_link": true
},
"gpuClass": "standard",