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": [], "outputs": [],
"source": [ "source": [
"! git clone https://github.com/kuprel/min-dalle\n", "%%shell\n",
"! mkdir -p /content/min-dalle/pretrained/vqgan/\n", "\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", "git clone https://github.com/kuprel/min-dalle\n",
"! pip install torch flax==0.4.2 wandb\n", "mkdir -p /content/min-dalle/pretrained/vqgan/\n",
"! wandb login --anonymously\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",
"! wandb artifact get --root=/content/min-dalle/pretrained/dalle_bart_mini dalle-mini/dalle-mini/mini-1:v0\n", "pip install torch flax==0.4.2 wandb\n",
"! wandb artifact get --root=/content/min-dalle/pretrained/dalle_bart_mega dalle-mini/dalle-mini/mega-1-fp16:v14\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/" "base_uri": "https://localhost:8080/"
} }
}, },
"execution_count": 2, "execution_count": null,
"outputs": [ "outputs": [
{ {
"output_type": "stream", "output_type": "stream",
@ -101,7 +103,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": 5, "execution_count": null,
"metadata": { "metadata": {
"colab": { "colab": {
"base_uri": "https://localhost:8080/", "base_uri": "https://localhost:8080/",
@ -169,7 +171,7 @@
"name": "min-dalle", "name": "min-dalle",
"provenance": [], "provenance": [],
"machine_shape": "hm", "machine_shape": "hm",
"authorship_tag": "ABX9TyMD0M+1bhwaISRugJcCWvwm", "authorship_tag": "ABX9TyM7nP04EHVNWqYy09gjYVDl",
"include_colab_link": true "include_colab_link": true
}, },
"gpuClass": "standard", "gpuClass": "standard",