Commit Graph

271 Commits

Author SHA1 Message Date
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
Brett Kuprel
fb97ba5e20 update readme, cleanup 2022-06-30 07:41:31 -04:00
Brett Kuprel
1e18ba0ffa is_expendable argument reduces memory usage for command line script 2022-06-30 06:43:10 -04:00
Brett Kuprel
38377107da
Update setup.sh 2022-06-29 21:45:05 -04:00
Brett Kuprel
c76226571e
Update cog.yaml 2022-06-29 19:14:25 -04:00
Brett Kuprel
8b86152359
Update cog.yaml 2022-06-29 19:14:10 -04:00
Brett Kuprel
5f931f6899 simplified colab setup 2022-06-29 16:32:51 -04:00
Brett Kuprel
b7b6df23f7 updated replicate predict.py file 2022-06-29 15:53:25 -04:00
Brett Kuprel
005ee4938e
Update README.md 2022-06-29 15:24:09 -04:00
Brett Kuprel
9fbfe0b76e
Delete predict.py 2022-06-29 15:20:45 -04:00
Brett Kuprel
15b0c03485
Merge pull request #38 from chenxwh/replicate
Add Web Demo & Docker environment
2022-06-29 15:20:20 -04:00
Brett Kuprel
d99828a239 simplified flax attention and matched torch attention 2022-06-29 14:56:28 -04:00