Upload folder using huggingface_hub
Browse files- config.json +4 -4
- configuration_footballbert.py +4 -4
config.json
CHANGED
|
@@ -9,10 +9,10 @@
|
|
| 9 |
"max_position_embeddings": 68,
|
| 10 |
"model_type": "footballbert",
|
| 11 |
"auto_map": {
|
| 12 |
-
"AutoConfig": "FootballBERTConfig",
|
| 13 |
-
"AutoModel": "FootballBERTModel",
|
| 14 |
-
"AutoModelForMaskedLM": "FootballBERTForMaskedPlayerPrediction",
|
| 15 |
-
"AutoTokenizer": "FootballBERTTokenizer"
|
| 16 |
},
|
| 17 |
"num_attention_heads": 1,
|
| 18 |
"num_hidden_layers": 6,
|
|
|
|
| 9 |
"max_position_embeddings": 68,
|
| 10 |
"model_type": "footballbert",
|
| 11 |
"auto_map": {
|
| 12 |
+
"AutoConfig": "footballbert_hf.FootballBERTConfig",
|
| 13 |
+
"AutoModel": "footballbert_hf.FootballBERTModel",
|
| 14 |
+
"AutoModelForMaskedLM": "footballbert_hf.FootballBERTForMaskedPlayerPrediction",
|
| 15 |
+
"AutoTokenizer": "footballbert_hf.FootballBERTTokenizer"
|
| 16 |
},
|
| 17 |
"num_attention_heads": 1,
|
| 18 |
"num_hidden_layers": 6,
|
configuration_footballbert.py
CHANGED
|
@@ -76,8 +76,8 @@ class FootballBERTConfig(PretrainedConfig):
|
|
| 76 |
|
| 77 |
# auto mapping for model
|
| 78 |
self.auto_map = {
|
| 79 |
-
"AutoConfig": "FootballBERTConfig",
|
| 80 |
-
"AutoModel": "FootballBERTModel",
|
| 81 |
-
"AutoModelForMaskedLM": "FootballBERTForMaskedPlayerPrediction",
|
| 82 |
-
"AutoTokenizer": "FootballBERTTokenizer"
|
| 83 |
}
|
|
|
|
| 76 |
|
| 77 |
# auto mapping for model
|
| 78 |
self.auto_map = {
|
| 79 |
+
"AutoConfig": "footballbert_hf.FootballBERTConfig",
|
| 80 |
+
"AutoModel": "footballbert_hf.FootballBERTModel",
|
| 81 |
+
"AutoModelForMaskedLM": "footballbert_hf.FootballBERTForMaskedPlayerPrediction",
|
| 82 |
+
"AutoTokenizer": "footballbert_hf.FootballBERTTokenizer"
|
| 83 |
}
|