systemk
/

dsir-wiki-ja-5k

Feature Extraction

Model card Files Files and versions Community

minato-ryan commited on Mar 14

Commit

2618a43

•

1 Parent(s): c65c668

Update config.py

Files changed (1) hide show

config.py +2 -1

config.py CHANGED Viewed

@@ -12,8 +12,9 @@ class DSIRConfig(PretrainedConfig):
         n: int = 2,
         num_buckets: int = 10_000,
         laplace_smoothing: float = 1e-4,
     ):
-        super().__init__()
         self.n = n
         self.num_buckets = num_buckets
         self.laplace_smoothing = laplace_smoothing

         n: int = 2,
         num_buckets: int = 10_000,
         laplace_smoothing: float = 1e-4,
+        **kwargs
     ):
+        super().__init__(**kwargs)
         self.n = n
         self.num_buckets = num_buckets
         self.laplace_smoothing = laplace_smoothing