Commit Graph

9 Commits

Author SHA1 Message Date
Hermes
893fb0950d Update to burn v0.14.0 and switch to .mpk model file 2024-10-05 14:19:49 -04:00
Gadersd
01b1aea897 Add custom backend to enable flash attention 2023-09-07 12:54:27 -04:00
Gadersd
f4c58c1790 Replace helper functions with native burn functions 2023-09-07 12:23:18 -04:00
Gadersd
1830756917 Use wgpu by default and ndarray for convert 2023-08-08 15:32:21 -04:00
Gadersd
6e065035cd Add batching to generate two images at a time 2023-08-05 19:41:11 -04:00
Gadersd
ce91c8838f Fix model initialization bugs 2023-08-05 16:20:49 -04:00
Gadersd
9e247777fa Add model loading, saving, and conversion functionality 2023-08-05 10:14:03 -04:00
Gadersd
77f30aefa7 Add first successful sampling implementation 2023-08-04 17:01:44 -04:00
Gadersd
e4145441eb Add files via upload
Add initial project files
2023-08-04 14:32:47 -04:00