huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-09-11 04:14:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add missing cuda prefix

Browse Source 
Signed-off-by: Adrien Gallouët <angt@huggingface.co>
This commit is contained in:
Adrien Gallouët 2025-02-07 11:48:16 +00:00
parent 4841f71a0e
commit b6cfa0fbc0
No known key found for this signature in database
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
backends/llamacpp/build.rs
View File

@ -39,7 +39,8 @@ fn main() {
if version.len() > 2 {
version.pop();
}
pkg_config::Config::new().probe(&version.join(".")).unwrap();
let cuda_version = format!("cuda-{}", version.join("."));
pkg_config::Config::new().probe(&cuda_version).unwrap();
}
pkg_config::Config::new().probe("llama").unwrap();