mirror of
https://github.com/huggingface/transformers.git
synced 2025-07-15 10:38:23 +06:00
31 lines
723 B
Python
31 lines
723 B
Python
dependencies = ['torch', 'tqdm', 'boto3', 'requests', 'regex']
|
|
|
|
from hubconfs.bert_hubconf import (
|
|
bertTokenizer,
|
|
bertModel,
|
|
bertForNextSentencePrediction,
|
|
bertForPreTraining,
|
|
bertForMaskedLM,
|
|
bertForSequenceClassification,
|
|
bertForMultipleChoice,
|
|
bertForQuestionAnswering,
|
|
bertForTokenClassification
|
|
)
|
|
from hubconfs.gpt_hubconf import (
|
|
openAIGPTTokenizer,
|
|
openAIGPTModel,
|
|
openAIGPTLMHeadModel,
|
|
openAIGPTDoubleHeadsModel
|
|
)
|
|
from hubconfs.gpt2_hubconf import (
|
|
gpt2Tokenizer,
|
|
gpt2Model,
|
|
gpt2LMHeadModel,
|
|
gpt2DoubleHeadsModel
|
|
)
|
|
from hubconfs.transformer_xl_hubconf import (
|
|
transformerXLTokenizer,
|
|
transformerXLModel,
|
|
transformerXLLMHeadModel
|
|
)
|