deba7abc21
Merge pull request 'Vibed Changes' ( #1 ) from test into main
...
Reviewed-on: #1
2026-03-03 22:17:25 +01:00
90e0f4c985
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:17:25 +01:00
rasmus
6a73e7b27e
updated to the newest version of burn
2026-03-03 22:17:25 +01:00
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
7531196def
Delete LICENSE
2026-03-03 22:16:48 +01:00
fc2c8985d1
Delete README.md
2026-03-03 22:16:43 +01:00
3ce6a57b4e
Initial commit
2026-03-03 22:12:25 +01:00
