30 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
9e4d7bd310 Support is hopeless 2023-09-17 12:43:26 -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
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
b87273c2be Update README.md 2023-08-07 10:59:57 -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
1d1272b91f Update README.md 2023-08-06 16:28:03 -04:00
Gadersd
562b06461f Use cpu for weight dumping 2023-08-06 13:55:20 -04:00
Gadersd
daba33ebf9 Add wgpu option 2023-08-06 13:28:16 -04:00
Gadersd
8a76c234e7 Update README.md
Add support info
2023-08-05 20:18:22 -04:00
Gadersd
00c7ac9b4b Update README.md 2023-08-05 19:48:57 -04:00
Gadersd
6e065035cd Add batching to generate two images at a time 2023-08-05 19:41:11 -04:00
Gadersd
57b446c08d Add example image 2023-08-05 19:06:34 -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
c408de50b7 Update README 2023-08-05 10:39:51 -04:00
Gadersd
9e247777fa Add model loading, saving, and conversion functionality 2023-08-05 10:14:03 -04:00
Gadersd
41fce2a47e Remove stable_diffusion.py 2023-08-04 21:22:47 -04:00
Gadersd
153a545406 Clean up, add attribution, and change name to dump.py 2023-08-04 20:05:45 -04:00
Gadersd
77f30aefa7 Add first successful sampling implementation 2023-08-04 17:01:44 -04:00
Gadersd
3bf5b8c477 Create .gitignore 2023-08-04 14:34:36 -04:00
Gadersd
e4145441eb Add files via upload
Add initial project files
2023-08-04 14:32:47 -04:00
Gadersd
1aed8b655a Initial commit 2023-08-04 14:28:05 -04:00