Upload model
Browse files- configuration.py +3 -2
configuration.py
CHANGED
|
@@ -9,9 +9,10 @@ class WangchanbertaEncoderConfig(PretrainedConfig):
|
|
| 9 |
th_model_base: str = "airesearch/wangchanberta-base-att-spm-uncased",
|
| 10 |
input_text_embedding_dim: int = 768,
|
| 11 |
output_embedding_dim: int = 512,
|
| 12 |
-
dropout: float = 0.2
|
|
|
|
| 13 |
):
|
| 14 |
-
super().__init__()
|
| 15 |
self.th_model_base = th_model_base
|
| 16 |
self.input_text_embedding_dim = input_text_embedding_dim
|
| 17 |
self.output_embedding_dim = output_embedding_dim
|
|
|
|
| 9 |
th_model_base: str = "airesearch/wangchanberta-base-att-spm-uncased",
|
| 10 |
input_text_embedding_dim: int = 768,
|
| 11 |
output_embedding_dim: int = 512,
|
| 12 |
+
dropout: float = 0.2,
|
| 13 |
+
**kwargs,
|
| 14 |
):
|
| 15 |
+
super().__init__(**kwargs)
|
| 16 |
self.th_model_base = th_model_base
|
| 17 |
self.input_text_embedding_dim = input_text_embedding_dim
|
| 18 |
self.output_embedding_dim = output_embedding_dim
|