Fast CPU kernel for transposed 1d convolutions. (#1822)

* Fast CPU kernel for transposed 1d convolutions.

* Bugfix.
This commit is contained in:
Laurent Mazare
2024-03-08 22:43:07 +01:00
committed by GitHub
parent e7fc1daa21
commit 3440cec3a0
3 changed files with 100 additions and 22 deletions

View File

@ -120,7 +120,7 @@ fn main() -> Result<()> {
Some(w) => std::path::PathBuf::from(w),
None => repo.get("first_stage.safetensors")?,
};
let second_stage_weights = match &args.first_stage_weights {
let second_stage_weights = match &args.second_stage_weights {
Some(w) => std::path::PathBuf::from(w),
None => repo.get("second_stage.safetensors")?,
};