gonglinyuan
/

metro_t0pp_largepp

Text2Text Generation

Model card Files Files and versions Community

gonglinyuan commited on Aug 18, 2023

Commit

599a88b

·

1 Parent(s): 561d359

Update sentencepiece_bpe.py

Files changed (1) hide show

sentencepiece_bpe.py +1 -1

sentencepiece_bpe.py CHANGED Viewed

@@ -28,7 +28,7 @@ class SentencepieceConfig:
 class SentencepieceBPE(object):
     def __init__(self, cfg):
-        cfg = SentencepieceConfig(cfg)
         self.enable_sampling = cfg.sentencepiece_enable_sampling
         self.alpha = cfg.sentencepiece_alpha
         sentencepiece_model = cfg.sentencepiece_model

 class SentencepieceBPE(object):
     def __init__(self, cfg):
+        cfg = SentencepieceConfig(**cfg)
         self.enable_sampling = cfg.sentencepiece_enable_sampling
         self.alpha = cfg.sentencepiece_alpha
         sentencepiece_model = cfg.sentencepiece_model