Update vicuna-chat.rs

Dognam · May 12, 2023 · da3e390 · da3e390
1 parent e6fb66b
commit da3e390
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/crates/llm/examples/vicuna-chat.rs b/crates/llm/examples/vicuna-chat.rs
@@ -132,6 +132,11 @@ fn load_progress_callback(
                 prev_load_time = std::time::Instant::now();
             }
         }
+        LoadProgress::LoraApplied { name } => {
+            if let Some(sp) = sp.as_mut() {
+                sp.update_text(format!("Applied Lora: {}", name));
+            };
+        }
         LoadProgress::Loaded {
             file_size,
             tensor_count,