Merge branch 'main' into zero-shot-roberta

guillaume-be · Sep 26, 2023 · 0fcf5ab · 0fcf5ab
2 parents 49b7a5d + 9575902
commit 0fcf5ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pipelines/zero_shot_classification.rs b/src/pipelines/zero_shot_classification.rs
@@ -643,7 +643,7 @@ impl ZeroShotClassificationModel {
     /// # Ok(())
     /// # }
     /// ```
-    fn new_with_tokenizer(
+    pub fn new_with_tokenizer(
         config: ZeroShotClassificationConfig,
         tokenizer: TokenizerOption,
     ) -> Result<ZeroShotClassificationModel, RustBertError> {