mirror of
https://github.com/huggingface/candle.git
synced 2025-06-19 03:54:56 +00:00
A few more tweaks.
This commit is contained in:
@ -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")?,
|
||||
};
|
||||
|
Reference in New Issue
Block a user