Upload modeling_ankh.py with huggingface_hub
Browse files- modeling_ankh.py +1 -0
modeling_ankh.py
CHANGED
|
@@ -784,6 +784,7 @@ class AnkhMaskedLMOutput(ModelOutput):
|
|
| 784 |
|
| 785 |
class FastAnkhConfig(PretrainedConfig):
|
| 786 |
model_type = "fast_ankh"
|
|
|
|
| 787 |
|
| 788 |
def __init__(
|
| 789 |
self,
|
|
|
|
| 784 |
|
| 785 |
class FastAnkhConfig(PretrainedConfig):
|
| 786 |
model_type = "fast_ankh"
|
| 787 |
+
attribute_map = {"hidden_size": "d_model"}
|
| 788 |
|
| 789 |
def __init__(
|
| 790 |
self,
|