Brett Kuprel
|
d99828a239
|
simplified flax attention and matched torch attention
|
2022-06-29 14:56:28 -04:00 |
|
Brett Kuprel
|
ed91ab4a30
|
refactored to load models once and run multiple times
|
2022-06-29 09:42:12 -04:00 |
|
Brett Kuprel
|
34df2b97df
|
previous commit broke colab example, so adjusting flax requirement to 0.4.2 for now
|
2022-06-28 08:04:08 -04:00 |
|
Brett Kuprel
|
38ebe54a38
|
works with latest flax version 0.5.2 now
|
2022-06-28 07:12:29 -04:00 |
|
Brett Kuprel
|
18e6a9852f
|
license and cleanup
|
2022-06-27 14:34:10 -04:00 |
|
Brett Kuprel
|
018414a5c3
|
fixed relative imports
|
2022-06-27 12:43:47 -04:00 |
|