Commit Graph

14 Commits

Author SHA1 Message Date
Gadersd
ccbf062514 Add custom backend to enable flash attention 2026-03-03 22:17:25 +01:00
Gadersd
32a3ad9b3c Replace helper functions with native burn functions 2026-03-03 22:17:25 +01:00
Gadersd
d4afd71fda Use wgpu by default and ndarray for convert 2026-03-03 22:17:25 +01:00
Gadersd
68d4310aa8 Switch to burn 0.9.0 to gain fast model io without the need for a custom recorder 2026-03-03 22:17:25 +01:00
Gadersd
c596eab0e3 Add wgpu option 2026-03-03 22:17:25 +01:00
Gadersd
070469d721 Add batching to generate two images at a time 2026-03-03 22:17:25 +01:00
Gadersd
ebf7c32e88 Fix allocation bug 2026-03-03 22:17:25 +01:00
Gadersd
a84825838d Use buffer to increase model loading/saving speed tremendously 2026-03-03 22:17:25 +01:00
Gadersd
c144649e1d Fix model initialization bugs 2026-03-03 22:17:25 +01:00
Gadersd
bb80c045e0 Correct a few trivial bugs 2026-03-03 22:17:25 +01:00
Gadersd
6f9f4f1f61 Add ability to load dump or burn model in sample binary 2026-03-03 22:17:25 +01:00
Gadersd
c774f80f13 Add model loading, saving, and conversion functionality 2026-03-03 22:17:25 +01:00
Gadersd
8e7a8d9be4 Add first successful sampling implementation 2026-03-03 22:17:25 +01:00
Gadersd
7a8d13d23c Add files via upload
Add initial project files
2026-03-03 22:17:25 +01:00