Commit Graph

132 Commits

Author SHA1 Message Date
Brett Kuprel
70bd3cb80d update examples 2022-07-01 14:28:51 -04:00
Brett Kuprel
73d49af07f update examples 2022-07-01 14:26:36 -04:00
Brett Kuprel
763018e212 fix colab setup 2022-07-01 14:23:22 -04:00
Brett Kuprel
867f7d6970 update readme 2022-07-01 14:19:35 -04:00
Brett Kuprel
07ce93d5f8 moved flax model and conversion code to separate repository 2022-07-01 14:06:50 -04:00
Brett Kuprel
febd18df77 update readme 2022-07-01 12:55:30 -04:00
Brett Kuprel
b61041f766 add back requirements.txt 2022-07-01 12:53:55 -04:00
Brett Kuprel
f71bd054fd Merge branch 'main' of https://github.com/kuprel/min-dalle 2022-07-01 12:21:39 -04:00
Brett Kuprel
ee846ba1cf update setup scripts 2022-07-01 12:21:31 -04:00
Brett Kuprel
e12314b387 updated colab 2022-07-01 12:14:31 -04:00
Brett Kuprel
85f5866eff remove config.json dependency, default to torch in image_from_text.py 2022-07-01 12:03:37 -04:00
Brett Kuprel
4404e70764 fixed setup_torch.sh 2022-07-01 11:38:11 -04:00
Brett Kuprel
5cc7d0038f fixed setup_flax.sh 2022-07-01 11:32:36 -04:00
Brett Kuprel
2a4e3db15b update setup_torch.sh to download mini model too 2022-07-01 11:27:26 -04:00
Brett Kuprel
ff77e96b07 update readme 2022-07-01 11:19:04 -04:00
Brett Kuprel
b785accffd update readme 2022-07-01 11:18:30 -04:00
Brett Kuprel
cc90fa4a90 download pre-converted torch files in colab 2022-07-01 11:09:49 -04:00
Brett Kuprel
09a0f85b8e separate setup processes for flax and torch 2022-07-01 11:08:33 -04:00
Brett Kuprel
7bf76deafb fixed wrong file path 2022-07-01 10:58:29 -04:00
Brett Kuprel
fffd0f2b83 Merge branch 'main' of https://github.com/kuprel/min-dalle 2022-07-01 10:17:40 -04:00
Brett Kuprel
e4c2be54cb save converted detokenizer params 2022-07-01 10:17:29 -04:00
Brett Kuprel
26c336bd4d
Merge pull request #20 from Ewpratten/patch-1
Explicitly call wandb as a python module.
2022-07-01 07:55:48 -04:00
Brett Kuprel
8b5960b687 update setup.sh 2022-07-01 07:54:19 -04:00
Brett Kuprel
d13e573fb8 mega works with latest flax version 0.5.2 now, removing 0.4.2 pin 2022-07-01 03:07:32 -04:00
Brett Kuprel
b40fd83a0d mega works with latest flax version 0.5.2 now, removing 0.4.2 pin 2022-07-01 02:58:43 -04:00
Brett Kuprel
e3329a7f64
Merge branch 'main' into patch-1 2022-07-01 02:40:38 -04:00
Brett Kuprel
9ac9c0ca30 simplified colab 2022-06-30 21:23:15 -04:00
Brett Kuprel
eaee59a1ef update readme 2022-06-30 21:05:02 -04:00
Brett Kuprel
e9ff397009 Merge branch 'main' of https://github.com/kuprel/min-dalle 2022-06-30 21:04:06 -04:00
Brett Kuprel
f8bffc6892 update readme 2022-06-30 21:03:56 -04:00
Brett Kuprel
b93c4fd0e4 simplified colab 2022-06-30 21:02:27 -04:00
Brett Kuprel
432ffa8d8c reusable mega torch model works in standard colab runtime 2022-06-30 20:55:42 -04:00
Brett Kuprel
08b158d580 updated readme 2022-06-30 16:50:04 -04:00
Brett Kuprel
2311a1af7b delete cache 2022-06-30 15:48:20 -04:00
Brett Kuprel
d9d7f34b22 update readme 2022-06-30 15:33:53 -04:00
Brett Kuprel
23620b95cd Merge branch 'main' of https://github.com/kuprel/min-dalle 2022-06-30 15:19:21 -04:00
Brett Kuprel
c8f3304363 update readme 2022-06-30 15:18:29 -04:00
Brett Kuprel
a3d81888b5 generate avocado armchair with pytorch in standard colab runtime 2022-06-30 15:17:35 -04:00
Brett Kuprel
ed0193296c mega model works in standard colab runtime 2022-06-30 14:57:53 -04:00
Brett Kuprel
b913b58353 pre converting params to torch allows mega to run in standard colab runtime 2022-06-30 14:54:08 -04:00
Brett Kuprel
de97fcf06b
Update requirements.txt 2022-06-30 13:13:33 -04:00
Brett Kuprel
c2a3858c96 delete params sooner 2022-06-30 11:44:36 -04:00
Brett Kuprel
f951424e38 is_reusable 2022-06-30 11:25:24 -04:00
Brett Kuprel
3e64e868ef Merge branch 'main' of https://github.com/kuprel/min-dalle 2022-06-30 11:09:17 -04:00
Brett Kuprel
b55bcba4c0 removed deepcopy, delete expendable parameters after use 2022-06-30 11:09:09 -04:00
Brett Kuprel
b44e8997a0 still some issues with mega in standard runtime, reverting for now 2022-06-30 10:02:08 -04:00
Brett Kuprel
6d1ac07442 mega model in standard runtime 2022-06-30 09:43:13 -04:00
Brett Kuprel
e9c01e32a5 mega model works with standard colab runtime in expendable mode 2022-06-30 09:39:14 -04:00
Brett Kuprel
41a44068d0 keep params in expendable mode 2022-06-30 09:36:32 -04:00
Brett Kuprel
df9aa6f915 sort -> topk, prev_token_and_index -> prev_token, token_index 2022-06-30 09:04:11 -04:00