Commit Graph

32 Commits

Author SHA1 Message Date
Hermes 75f0cedd9f Update to burn v0.14.0 and switch to .mpk model file 2026-03-03 22:17:25 +01:00
Gadersd 3c49b0a151 Support is hopeless 2026-03-03 22:17:25 +01:00
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 167e45fc30 Use torch by default 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 0101e8f930 Update README.md 2026-03-03 22:17:25 +01:00
Gadersd effd3916b5 Update Cargo.toml
Remove unnecessary dependency
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 1b5426c887 Update README.md 2026-03-03 22:17:25 +01:00
Gadersd 2b1a5a6f78 Use cpu for weight dumping 2026-03-03 22:17:25 +01:00
Gadersd c596eab0e3 Add wgpu option 2026-03-03 22:17:25 +01:00
Gadersd 1373918c2e Update README.md
Add support info
2026-03-03 22:17:25 +01:00
Gadersd a92e5e760c Update README.md 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 b6a998c960 Add example image 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 bcf6b01d3a Update README 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 3c2c43dd93 Remove stable_diffusion.py 2026-03-03 22:17:25 +01:00
Gadersd 63bc2ffbd9 Clean up, add attribution, and change name to dump.py 2026-03-03 22:17:25 +01:00
Gadersd 8e7a8d9be4 Add first successful sampling implementation 2026-03-03 22:17:25 +01:00
Gadersd b794e9a9ec Create .gitignore 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
Gadersd fb3bb40fba Initial commit 2026-03-03 22:17:25 +01:00
Ben_Kosytorz 7531196def Delete LICENSE 2026-03-03 22:16:48 +01:00
Ben_Kosytorz fc2c8985d1 Delete README.md 2026-03-03 22:16:43 +01:00
Ben_Kosytorz 3ce6a57b4e Initial commit 2026-03-03 22:12:25 +01:00