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

cleanup

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-01 14:35:13 -04:00
parent 70bd3cb80d
commit 2097c6ed5a
2 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md vendored
View File

@ -7,7 +7,7 @@ This is a minimal implementation of Boris Dayma's [DALL·E Mini](https://github.
It currently takes **7.4 seconds** to generate an image with DALL·E Mega with PyTorch on a standard GPU runtime in Colab
The flax model, and the code for coverting it to torch, has been moved [here](https://github.com/kuprel/min-dalle-flax).
The flax model, and the code for coverting it to torch, have been moved [here](https://github.com/kuprel/min-dalle-flax).
### Setup

7
min_dalle.ipynb vendored
View File

@ -84,20 +84,19 @@
"%%shell\n",
"\n",
"repo_path=\"https://huggingface.co/kuprel/min-dalle/resolve/main\"\n",
"\n",
"mega_path=\"/content/min-dalle/pretrained/dalle_bart_mega\"\n",
"vqgan_path=\"/content/min-dalle/pretrained/vqgan\"\n",
"\n",
"git clone https://github.com/kuprel/min-dalle\n",
"\n",
"mkdir -p ${vqgan_path}\n",
"mkdir -p ${mega_path}\n",
"mkdir -p ${vqgan_path}\n",
"\n",
"curl ${repo_path}/detoker.pt -L --output ${vqgan_path}/detoker.pt\n",
"curl ${repo_path}/vocab.json -L --output ${mega_path}/vocab.json\n",
"curl ${repo_path}/merges.txt -L --output ${mega_path}/merges.txt\n",
"curl ${repo_path}/encoder.pt -L --output ${mega_path}/encoder.pt\n",
"curl ${repo_path}/decoder.pt -L --output ${mega_path}/decoder.pt"
"curl ${repo_path}/decoder.pt -L --output ${mega_path}/decoder.pt\n",
"curl ${repo_path}/detoker.pt -L --output ${vqgan_path}/detoker.pt"
]
},
{