huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 18:28:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Pin the revision used for phi-v2 + make it the default. (#1572)

Browse Source 
* Pin the revision used for phi-v2 + make it the default.

* Tweak the custom-ops build.
This commit is contained in:
Laurent Mazare
2024-01-12 09:19:30 +01:00
committed by GitHub
parent 41915184bb
commit 6242276c09
2 changed files with 3 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
candle-examples/build.rs
View File

@ -27,11 +27,5 @@ fn main() -> Result<()> {
bindings.write(kdir.rust_target).unwrap() bindings.write(kdir.rust_target).unwrap()
} }
} }
#[cfg(not(feature = "cuda"))]
{
for kdir in KERNEL_DIRS.iter() {
let _file = std::fs::File::create(kdir.rust_target)?;
}
}
Ok(()) Ok(())
} }

7
candle-examples/examples/phi/main.rs
View File

@ -169,7 +169,7 @@ struct Args {
#[arg(long)] #[arg(long)]
model_id: Option<String>, model_id: Option<String>,
#[arg(long, default_value = "1.5")] #[arg(long, default_value = "2")]
model: WhichModel, model: WhichModel,
#[arg(long)] #[arg(long)]
@ -247,9 +247,8 @@ fn main() -> Result<()> {
match args.model { match args.model {
WhichModel::V1 => "refs/pr/2".to_string(), WhichModel::V1 => "refs/pr/2".to_string(),
WhichModel::V1_5 => "refs/pr/18".to_string(), WhichModel::V1_5 => "refs/pr/18".to_string(),
WhichModel::V2 | WhichModel::PuffinPhiV2 | WhichModel::PhiHermes => { WhichModel::V2 => "834565c23f9b28b96ccbeabe614dd906b6db551a".to_string(),
"main".to_string() WhichModel::PuffinPhiV2 | WhichModel::PhiHermes => "main".to_string(),
}
} }
} }
} }