SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Fix imports in conversion scripts (#9674)

Browse Source
This commit is contained in:
Sylvain Gugger
2021-01-19 09:40:15 -05:00
committed by GitHub
parent 2390c16fd2
commit 053efc5d2d
28 changed files with 75 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/transformers/models/roberta/convert_roberta_original_pytorch_checkpoint_to_pytorch.py
View File

@@ -24,9 +24,15 @@ from fairseq.models.roberta import RobertaModel as FairseqRobertaModel
from fairseq.modules import TransformerSentenceEncoderLayer
from packaging import version
from ...models.bert.modeling_bert import BertIntermediate, BertLayer, BertOutput, BertSelfAttention, BertSelfOutput
from ...utils import logging
from .modeling_roberta import RobertaConfig, RobertaForMaskedLM, RobertaForSequenceClassification
from transformers import RobertaConfig, RobertaForMaskedLM, RobertaForSequenceClassification
from transformers.models.bert.modeling_bert import (
BertIntermediate,
BertLayer,
BertOutput,
BertSelfAttention,
BertSelfOutput,
)
from transformers.utils import logging
if version.parse(fairseq.__version__) < version.parse("0.9.0"):