Commit Graph

59 Commits

Author SHA1 Message Date
Brett Kuprel
cc9bdb8f7b update readme 2022-07-01 19:08:42 -04:00
Brett Kuprel
847daa5be2 update readme 2022-07-01 19:05:46 -04:00
Brett Kuprel
fb77728ea8 update readme 2022-07-01 19:03:18 -04:00
Brett Kuprel
dc0d02a7a7 update readme 2022-07-01 19:02:49 -04:00
Brett Kuprel
2614a83f35 update readme 2022-07-01 19:01:07 -04:00
Brett Kuprel
8f58a51a7d fix readme 2022-07-01 18:54:05 -04:00
Brett Kuprel
f1f433048c fix readme 2022-07-01 18:52:40 -04:00
Brett Kuprel
f86a4595e8 fix readme 2022-07-01 18:51:06 -04:00
Brett Kuprel
fff44d683e update readme, random seed if none specified 2022-07-01 18:50:11 -04:00
Brett Kuprel
ec9e22e4d4 code format 2022-07-01 18:32:48 -04:00
Brett Kuprel
7b5d972914 update readme 2022-07-01 18:30:22 -04:00
Brett Kuprel
be2beca7c0 added to pypi 2022-07-01 18:16:55 -04:00
Brett Kuprel
2097c6ed5a cleanup 2022-07-01 14:35:13 -04:00
Brett Kuprel
70bd3cb80d update examples 2022-07-01 14:28:51 -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
85f5866eff remove config.json dependency, default to torch in image_from_text.py 2022-07-01 12:03:37 -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
09a0f85b8e separate setup processes for flax and torch 2022-07-01 11:08:33 -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
eaee59a1ef update readme 2022-06-30 21:05:02 -04:00
Brett Kuprel
f8bffc6892 update readme 2022-06-30 21:03:56 -04:00
Brett Kuprel
08b158d580 updated readme 2022-06-30 16:50:04 -04:00
Brett Kuprel
d9d7f34b22 update readme 2022-06-30 15:33:53 -04:00
Brett Kuprel
c8f3304363 update readme 2022-06-30 15:18:29 -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
41a44068d0 keep params in expendable mode 2022-06-30 09:36:32 -04:00
Brett Kuprel
fb97ba5e20 update readme, cleanup 2022-06-30 07:41:31 -04:00
Brett Kuprel
005ee4938e
Update README.md 2022-06-29 15:24:09 -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
Chenxi
fcc17c895d
Merge branch 'kuprel:main' into replicate 2022-06-29 19:50:47 +01:00
Chenxi
eb9f4c6b3b replicate demo 2022-06-29 19:50:10 +01:00
Brett Kuprel
a4df279fd2 simplified attention and keys_values state resulted in decrease in inference time to 7.3 seconds (from ~10 seconds) 2022-06-29 13:56:29 -04:00
Brett Kuprel
764b5bbc0e works with latest flax version 0.5.2, updated requirements.txt 2022-06-29 11:46:19 -04:00
Brett Kuprel
c4f613c89f readme wording 2022-06-29 11:01:46 -04:00
Brett Kuprel
6046863805 readme wording 2022-06-29 11:00:38 -04:00
Brett Kuprel
2b552fe9db readme wording 2022-06-29 10:54:01 -04:00
Brett Kuprel
b7c2414c76 readme wording 2022-06-29 10:47:08 -04:00
Brett Kuprel
4e62f85ab9 readme wording 2022-06-29 10:45:46 -04:00
Brett Kuprel
53695e32f7 updated readme with torch examples 2022-06-29 10:43:46 -04:00
Brett Kuprel
ed91ab4a30 refactored to load models once and run multiple times 2022-06-29 09:42:12 -04:00
Brett Kuprel
1fbb209623 fixed bug with cuda in detokenizer 2022-06-28 22:02:35 -04:00
Brett Kuprel
aef24ea157 torch.no_grad(), cleanup 2022-06-28 12:16:44 -04:00
Omar Sanseviero
efa40ab321
Update README.md 2022-06-28 09:25:56 +02:00
Brett Kuprel
6260252348 update readme 2022-06-27 22:08:29 -04:00
kuprel
2ad7009a16
Update README.md 2022-06-27 21:11:27 -04:00