Upload sCT
Browse files
config.py
CHANGED
@@ -9,7 +9,6 @@ class sCTConfig(PretrainedConfig): # noqa: N801
|
|
9 |
model_type = "sCT"
|
10 |
|
11 |
def __init__(self, **kwargs): # type: ignore
|
12 |
-
super().__init__()
|
13 |
self.alphabet_size = kwargs.get("alphabet_size", 7)
|
14 |
self.pad_token_id = kwargs.get("pad_token_id", 5)
|
15 |
self.mask_token_id = kwargs.get("mask_token_id", 6)
|
@@ -52,6 +51,8 @@ class sCTConfig(PretrainedConfig): # noqa: N801
|
|
52 |
self.sigma_min: float = kwargs.get("sigma_min", 1.0)
|
53 |
self.sigma_max: float = kwargs.get("sigma_max", 10.0)
|
54 |
|
|
|
|
|
55 |
def __post_init__(self) -> None: # type: ignore # noqa: N807
|
56 |
"""
|
57 |
Checks that the given values are compatible.
|
|
|
9 |
model_type = "sCT"
|
10 |
|
11 |
def __init__(self, **kwargs): # type: ignore
|
|
|
12 |
self.alphabet_size = kwargs.get("alphabet_size", 7)
|
13 |
self.pad_token_id = kwargs.get("pad_token_id", 5)
|
14 |
self.mask_token_id = kwargs.get("mask_token_id", 6)
|
|
|
51 |
self.sigma_min: float = kwargs.get("sigma_min", 1.0)
|
52 |
self.sigma_max: float = kwargs.get("sigma_max", 10.0)
|
53 |
|
54 |
+
super().__init__(**kwargs)
|
55 |
+
|
56 |
def __post_init__(self) -> None: # type: ignore # noqa: N807
|
57 |
"""
|
58 |
Checks that the given values are compatible.
|