754810ca88
feat: migrate to pure Rust backend with WGPU/Vulkan support
...
- Remove libtorch dependency and related features
- Switch to ndarray backend as default for CPU execution
- Update README to reflect new WGPU/Vulkan backend usage
- Simplify device selection to use only CPU backend
- Enable WGPU backend via feature flag for GPU acceleration
2026-03-03 22:07:14 +01:00
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
