huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 03:54:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Expliciting how to enable cuda.

Browse Source
This commit is contained in:
Nicolas Patry
2023-07-14 17:08:05 +02:00
parent 2d5e952cf9
commit dcb4a9291e
2 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -24,6 +24,8 @@ cargo run --example llama --release
cargo run --example falcon --release
```
In order to use **CUDA** add `--features cuda` to the example command line.
## Features

2
candle-core/src/dummy_cuda_backend.rs
View File

@ -9,7 +9,7 @@ pub struct CudaStorage;
macro_rules! fail {
() => {
unimplemented!("cuda support has not been enabled")
unimplemented!("cuda support has not been enabled, add `cuda` feature to enable.")
};
}