Update modeling_firstname_gender.py
Browse files
modeling_firstname_gender.py
CHANGED
|
@@ -184,7 +184,8 @@ class FirstNameGenderForCausalLM(PreTrainedModel):
|
|
| 184 |
config_class = FirstNameGenderConfig
|
| 185 |
base_model_prefix = "model"
|
| 186 |
supports_gradient_checkpointing = False
|
| 187 |
-
|
|
|
|
| 188 |
def __init__(self, config):
|
| 189 |
super().__init__(config)
|
| 190 |
|
|
|
|
| 184 |
config_class = FirstNameGenderConfig
|
| 185 |
base_model_prefix = "model"
|
| 186 |
supports_gradient_checkpointing = False
|
| 187 |
+
all_tied_weights_keys = {}
|
| 188 |
+
|
| 189 |
def __init__(self, config):
|
| 190 |
super().__init__(config)
|
| 191 |
|