Unverified Commit f2e0d6c2 authored by Zhongdongming Dai's avatar Zhongdongming Dai Committed by GitHub
Browse files

feat(dynamo-run): make the model name to be the same as the HF repo name (#749)

parent 22cacbb1
......@@ -124,10 +124,7 @@ pub async fn run(
// If it's an HF repo download it
if let Some(inner_model_path) = model_path.as_ref() {
if !inner_model_path.exists() {
model_name = inner_model_path
.iter()
.next_back()
.map(|s| s.to_string_lossy().to_string());
model_name = Some(inner_model_path.display().to_string());
model_path = Some(hub::from_hf(inner_model_path).await?);
}
}
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment