Skip to content

Commit

Permalink
make ZeroShotClassificationModel::new_with_tokenizer public (guillaum…
Browse files Browse the repository at this point in the history
…e-be#421)

Co-authored-by: Charles Samuels <[email protected]>
  • Loading branch information
njaard and Charles Samuels authored Sep 26, 2023
1 parent 0a1c3df commit 9575902
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/pipelines/zero_shot_classification.rs
Original file line number Diff line number Diff line change
Expand Up @@ -643,7 +643,7 @@ impl ZeroShotClassificationModel {
/// # Ok(())
/// # }
/// ```
fn new_with_tokenizer(
pub fn new_with_tokenizer(
config: ZeroShotClassificationConfig,
tokenizer: TokenizerOption,
) -> Result<ZeroShotClassificationModel, RustBertError> {
Expand Down

0 comments on commit 9575902

Please sign in to comment.