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