diff --git a/candle-lora-transformers/src/falcon.rs b/candle-lora-transformers/src/falcon.rs index 2490f5c..45b1413 100644 --- a/candle-lora-transformers/src/falcon.rs +++ b/candle-lora-transformers/src/falcon.rs @@ -532,6 +532,9 @@ impl Falcon { &self.config } + /// Load a Falcon model which will be converted to a LoRA model. + /// + /// The `merge` parameter merges the weights. pub fn load( vb: VarBuilder, cfg: Config,