Transformers fast import part 2 (#9446)

* Main init work

* Add version

* Change from absolute to relative imports

* Fix imports

* One more typo

* More typos

* Styling

* Make quality script pass

* Add necessary replace in template

* Fix typos

* Spaces are ignored in replace for some reason

* Forgot one models.

* Fixes for import

Co-authored-by: LysandreJik <lysandre.debut@reseau.eseo.fr>

* Add documentation

* Styling

Co-authored-by: LysandreJik <lysandre.debut@reseau.eseo.fr>
This commit is contained in:
Sylvain Gugger
2021-01-07 09:36:14 -05:00
committed by GitHub
parent a400fe8931
commit 758ed3332b
56 changed files with 2426 additions and 1377 deletions

View File

@@ -28,8 +28,8 @@ import re
import tensorflow as tf
import torch
from transformers import BertConfig, BertModel
from transformers.utils import logging
from ...utils import logging
from . import BertConfig, BertModel
logging.set_verbosity_info()

View File

@@ -19,8 +19,8 @@ import argparse
import torch
from transformers import BertConfig, BertForPreTraining, load_tf_weights_in_bert
from transformers.utils import logging
from ...utils import logging
from . import BertConfig, BertForPreTraining, load_tf_weights_in_bert
logging.set_verbosity_info()

View File

@@ -22,7 +22,7 @@ import numpy as np
import tensorflow as tf
import torch
from transformers import BertModel
from . import BertModel
def convert_pytorch_checkpoint_to_tf(model: BertModel, ckpt_dir: str, model_name: str):