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.
|