Expose config for Rust API

EricLBuehler · Oct 12, 2024 · 8ab7d12 · 8ab7d12
1 parent 1958832
commit 8ab7d12
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/mistralrs/src/model.rs b/mistralrs/src/model.rs
@@ -151,4 +151,9 @@ impl Model {
 
  Ok(self.runner.get_sender()?.send(request).await?)
  }
+
+ /// Retrieve some information about this model.
+ pub fn config(&self) -> &MistralRsConfig {
+ self.runner.config()
+ }
 }