Commit Graph

16 Commits

Author SHA1 Message Date
rasmus
6cfd6db5a5 updated to the newest version of burn 2026-03-03 15:12:59 +01:00
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
c24d37df00 Switch to burn 0.9.0 to gain fast model io without the need for a custom recorder 2023-08-06 19:44:13 -04:00
Gadersd
daba33ebf9 Add wgpu option 2023-08-06 13:28:16 -04:00
Gadersd
6e065035cd Add batching to generate two images at a time 2023-08-05 19:41:11 -04:00
Gadersd
058dd1d7cc Fix allocation bug 2023-08-05 17:05:20 -04:00
Gadersd
2f3d94f9ad Use buffer to increase model loading/saving speed tremendously 2023-08-05 16:48:26 -04:00
Gadersd
ce91c8838f Fix model initialization bugs 2023-08-05 16:20:49 -04:00
Gadersd
10ce5ac89d Correct a few trivial bugs 2023-08-05 13:46:14 -04:00
Gadersd
08f7347ae5 Add ability to load dump or burn model in sample binary 2023-08-05 13:27:37 -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