Skip to content

GitLab

"...resnet50_tensorflow.git" did not exist on "580aa3f6c0864d778c822066cdccbba7f40da100"
Commit e184ad13 authored by Julien Chaumond's avatar Julien Chaumond
Browse files

Close #2392

parent dfe012ad
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 7 deletions
src/transformers/pipelines.py
View file @ e184ad13
...@@ -883,10 +883,10 @@ SUPPORTED_TASKS = { ...@@ -883,10 +883,10 @@ SUPPORTED_TASKS = {
"pt": AutoModelForTokenClassification if is_torch_available() else None, "pt": AutoModelForTokenClassification if is_torch_available() else None,
"default": { "default": {
"model": { "model": {
"pt": "https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-cased-finetuned-conll03-english-pytorch_model.bin", "pt": "dbmdz/bert-large-cased-finetuned-conll03-english",
"tf": "https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-cased-finetuned-conll03-english-tf_model.h5", "tf": "dbmdz/bert-large-cased-finetuned-conll03-english",
}, },
"config": "https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-cased-finetuned-conll03-english-config.json", "config": "dbmdz/bert-large-cased-finetuned-conll03-english",
"tokenizer": "bert-large-cased", "tokenizer": "bert-large-cased",
}, },
}, },
...@@ -925,6 +925,7 @@ def pipeline( ...@@ -925,6 +925,7 @@ def pipeline(
pipeline('sentiment-analysis') pipeline('sentiment-analysis')
pipeline('question-answering', model='distilbert-base-uncased-distilled-squad', tokenizer='bert-base-cased') pipeline('question-answering', model='distilbert-base-uncased-distilled-squad', tokenizer='bert-base-cased')
pipeline('ner', model=AutoModel.from_pretrained(...), tokenizer=AutoTokenizer.from_pretrained(...) pipeline('ner', model=AutoModel.from_pretrained(...), tokenizer=AutoTokenizer.from_pretrained(...)
pipeline('ner', model='dbmdz/bert-large-cased-finetuned-conll03-english', tokenizer='bert-base-cased')
pipeline('ner', model='https://...pytorch-model.bin', config='https://...config.json', tokenizer='bert-base-cased') pipeline('ner', model='https://...pytorch-model.bin', config='https://...config.json', tokenizer='bert-base-cased')
""" """
# Retrieve the task # Retrieve the task
......
tests/test_pipelines.py
View file @ e184ad13
...@@ -21,16 +21,16 @@ TF_QA_FINETUNED_MODELS = { ...@@ -21,16 +21,16 @@ TF_QA_FINETUNED_MODELS = {
TF_NER_FINETUNED_MODELS = { TF_NER_FINETUNED_MODELS = {
( (
"bert-base-cased", "bert-base-cased",
"https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-cased-finetuned-conll03-english-tf_model.h5", "dbmdz/bert-large-cased-finetuned-conll03-english",
"https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-cased-finetuned-conll03-english-config.json", "dbmdz/bert-large-cased-finetuned-conll03-english",
) )
} }
NER_FINETUNED_MODELS = { NER_FINETUNED_MODELS = {
( (
"bert-base-cased", "bert-base-cased",
"https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-cased-finetuned-conll03-english-pytorch_model.bin", "dbmdz/bert-large-cased-finetuned-conll03-english",
"https://s3.amazonaws.com/models.huggingface.co/bert/bert-large-cased-finetuned-conll03-english-config.json", "dbmdz/bert-large-cased-finetuned-conll03-english",
) )
} }
......
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