Commit Graph

11 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
a62795347f Use torch by default 2023-08-24 17:12:53 -04:00
Gadersd
1830756917 Use wgpu by default and ndarray for convert 2023-08-08 15:32:21 -04:00
Gadersd
31c24a82ef Update Cargo.toml
Remove unnecessary dependency
2023-08-06 19:55:00 -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
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
e4145441eb Add files via upload
Add initial project files
2023-08-04 14:32:47 -04:00