3e5cc12855
* rm tf/flax tests * more flax deletions * revert fixture change * reverted test that should not be deleted; rm tf/flax test * revert * fix a few add-model-like tests * fix add-model-like checkpoint source * a few more * test_get_model_files_only_pt fix * fix test_retrieve_info_for_model_with_xxx * fix test_retrieve_model_classes * relative paths are the devil * add todo
20 lines
504 B
JSON
20 lines
504 B
JSON
{
|
|
"add_copied_from": true,
|
|
"old_model_type": "distilbert",
|
|
"new_model_patterns": {
|
|
"model_name": "BERT New",
|
|
"checkpoint": "huggingface/bert-new-base",
|
|
"model_type": "bert-new",
|
|
"model_lower_cased": "bert_new",
|
|
"model_camel_cased": "BertNew",
|
|
"model_upper_cased": "BERT_NEW",
|
|
"config_class": "BertNewConfig",
|
|
"tokenizer_class": "DistilBertTokenizer"
|
|
},
|
|
"frameworks": [
|
|
"pt",
|
|
"tf",
|
|
"flax"
|
|
]
|
|
}
|