refactor(stage-tamagotchi): rename
This commit is contained in:
@@ -11,7 +11,7 @@ impl ModelLoadProgressEmitterManager {
|
||||
Self { window }
|
||||
}
|
||||
|
||||
pub fn new_processor(
|
||||
pub fn new_for(
|
||||
self,
|
||||
filename: &str,
|
||||
) -> ModelLoadProgressEmitter {
|
||||
@@ -67,7 +67,11 @@ impl hf_hub::api::Progress for ModelLoadProgressEmitter {
|
||||
size: usize,
|
||||
) {
|
||||
self.size += size;
|
||||
self.progress = if self.total_size > 0 { (self.size as f32 / self.total_size as f32 * 100.0).min(100.0) } else { 100.0 };
|
||||
self.progress = if self.total_size > 0 {
|
||||
(self.size as f32 / self.total_size as f32 * 100.0).min(100.0)
|
||||
} else {
|
||||
100.0
|
||||
};
|
||||
self
|
||||
.window
|
||||
.emit(
|
||||
|
||||
@@ -115,16 +115,14 @@ impl WhisperProcessor {
|
||||
));
|
||||
|
||||
let config_filename =
|
||||
repo.download_with_progress("config.json", manager.clone().new_processor("config.json"))?;
|
||||
repo.download_with_progress("config.json", manager.clone().new_for("config.json"))?;
|
||||
println!("config_filename: {:?}", config_filename.display());
|
||||
let tokenizer_filename = repo.download_with_progress(
|
||||
"tokenizer.json",
|
||||
manager.clone().new_processor("tokenizer.json"),
|
||||
)?;
|
||||
let tokenizer_filename =
|
||||
repo.download_with_progress("tokenizer.json", manager.clone().new_for("tokenizer.json"))?;
|
||||
println!("tokenizer_filename: {:?}", tokenizer_filename.display());
|
||||
let model_filename = repo.download_with_progress(
|
||||
"model.safetensors",
|
||||
manager.clone().new_processor("model.safetensors"),
|
||||
manager.clone().new_for("model.safetensors"),
|
||||
)?;
|
||||
println!("model_filename: {:?}", model_filename.display());
|
||||
|
||||
|
||||
Reference in New Issue
Block a user