fix(tagger): model-neutral session log; changelog for JoyTag
create_tagger_session is shared by both models but logged "WD tagger: using CPU execution provider" even when loading JoyTag. Make the CPU/DirectML session logs model-neutral. Add the JoyTag tagging-model picker to the changelog now that it's runtime-verified.
This commit is contained in:
@@ -987,15 +987,13 @@ fn create_tagger_session(path: &Path, acceleration: TaggerAcceleration) -> Resul
|
||||
|
||||
let mut builder = match acceleration {
|
||||
TaggerAcceleration::Cpu => {
|
||||
log::info!(
|
||||
"WD tagger: using CPU execution provider ({intra_threads} intra-op threads)"
|
||||
);
|
||||
log::info!("Tagger: using CPU execution provider ({intra_threads} intra-op threads)");
|
||||
builder
|
||||
}
|
||||
TaggerAcceleration::Auto => builder
|
||||
.with_execution_providers([ep::DirectML::default().build().fail_silently()])
|
||||
.unwrap_or_else(|error| {
|
||||
log::info!("WD tagger: DirectML unavailable, falling back to CPU");
|
||||
log::info!("Tagger: DirectML unavailable, falling back to CPU");
|
||||
error.recover()
|
||||
}),
|
||||
TaggerAcceleration::Directml => builder
|
||||
|
||||
Reference in New Issue
Block a user