Skip to content

GitLab

"docs/git@developer.sourcefind.cn:zhaoyu6/sglang.git" did not exist on "5922c0cbf68a6e95cbca847cd8421b467ea6f0dd"
Commit 84a3a968 authored by LysandreJik's avatar LysandreJik
Browse files

Pytorch Hub & AutoModels

parent f6833963
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 27 deletions
hubconf.py
View file @ 84a3a968
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
from pytorch_transformers import (
AutoTokenizer, AutoModel, AutoModelWithLMHead, AutoModelForSequenceClassification, AutoModelForQuestionAnswering
)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment