A few more tweaks.

This commit is contained in:
laurent
2024-03-05 22:39:06 +01:00
parent 52ed77c16f
commit b5ee026cea
2 changed files with 3 additions and 7 deletions

View File

@ -188,7 +188,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")?,
};