feat: update workspace paths and enhance gitignore
- Updated stablediffusion crate path from "../stable-diffusion-burn" to "./crates/stable-diffusion-burn" for proper workspace resolution - Enhanced .gitignore to include generated model files (.mpk, .pt, .bin, .safetensors, .ckpt) and user_data directory - Added Cargo.lock to gitignore with appropriate comment - Reorganized IDE files section in gitignore for better clarity - Added newline at end of file for proper formatting
This commit is contained in:
30
crates/stable-diffusion-burn/burn-crates/burn-cuda/README.md
Normal file
30
crates/stable-diffusion-burn/burn-crates/burn-cuda/README.md
Normal file
@@ -0,0 +1,30 @@
|
||||
# Burn CUDA Backend
|
||||
|
||||
[Burn](https://github.com/tracel-ai/burn) CUDA backend
|
||||
|
||||
[](https://crates.io/crates/burn-cuda)
|
||||
[](https://github.com/tracel-ai/burn-cuda/blob/master/README.md)
|
||||
|
||||
This crate provides a CUDA backend for [Burn](https://github.com/tracel-ai/burn) using the
|
||||
[cubecl](https://github.com/tracel-ai/cubecl.git) and [cudarc](https://github.com/coreylowman/cudarc.git)
|
||||
crates.
|
||||
|
||||
## Usage Example
|
||||
|
||||
```rust
|
||||
#[cfg(feature = "cuda")]
|
||||
mod cuda {
|
||||
use burn_autodiff::Autodiff;
|
||||
use burn_cuda::{Cuda, CudaDevice};
|
||||
use mnist::training;
|
||||
|
||||
pub fn run() {
|
||||
let device = CudaDevice::default();
|
||||
training::run::<Autodiff<Cuda<f32, i32>>>(device);
|
||||
}
|
||||
}
|
||||
```
|
||||
|
||||
## Dependencies
|
||||
|
||||
Requires CUDA 12.x to be installed and on the `PATH`.
|
||||
Reference in New Issue
Block a user