Derive clone for TrainerWrapper (#1317)

This commit is contained in:
Jonatan Kłosko
2023-08-07 15:15:10 +02:00
committed by GitHub
parent efea6c7246
commit d47d3e377c
2 changed files with 2 additions and 2 deletions

View File

@ -140,7 +140,7 @@ impl Model for ModelWrapper {
}
}
#[derive(Serialize, Deserialize)]
#[derive(Clone, Serialize, Deserialize)]
pub enum TrainerWrapper {
BpeTrainer(BpeTrainer),
WordPieceTrainer(WordPieceTrainer),

View File

@ -88,7 +88,7 @@ impl WordPieceTrainerBuilder {
}
/// Trains a `WordPiece` model.
#[derive(Default, Deserialize, Serialize)]
#[derive(Default, Clone, Deserialize, Serialize)]
pub struct WordPieceTrainer {
bpe_trainer: BpeTrainer,
}