Add docs to LoRA falcon model

EricLBuehler · Oct 27, 2023 · f7701cc · f7701cc
1 parent ee80475
commit f7701cc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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,