Skip to content

GitLab

Commit d8b43600 authored by VictorSanh's avatar VictorSanh Committed by Victor SANH
Browse files

omission

parent ee5a6856
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/test_pipelines.py
View file @ d8b43600
...@@ -10,13 +10,13 @@ from .utils import require_tf, require_torch ...@@ -10,13 +10,13 @@ from .utils import require_tf, require_torch
QA_FINETUNED_MODELS = { QA_FINETUNED_MODELS = {
("bert-base-uncased", "bert-large-uncased-whole-word-masking-finetuned-squad", None), ("bert-base-uncased", "bert-large-uncased-whole-word-masking-finetuned-squad", None),
("bert-base-cased", "bert-large-cased-whole-word-masking-finetuned-squad", None), ("bert-base-cased", "bert-large-cased-whole-word-masking-finetuned-squad", None),
("bert-base-uncased", "distilbert-base-cased-distilled-squad", None), ("bert-base-cased", "distilbert-base-cased-distilled-squad", None),
} }
TF_QA_FINETUNED_MODELS = { TF_QA_FINETUNED_MODELS = {
("bert-base-uncased", "bert-large-uncased-whole-word-masking-finetuned-squad", None), ("bert-base-uncased", "bert-large-uncased-whole-word-masking-finetuned-squad", None),
("bert-base-cased", "bert-large-cased-whole-word-masking-finetuned-squad", None), ("bert-base-cased", "bert-large-cased-whole-word-masking-finetuned-squad", None),
("bert-base-uncased", "distilbert-base-cased-distilled-squad", None), ("bert-base-cased", "distilbert-base-cased-distilled-squad", None),
} }
TF_NER_FINETUNED_MODELS = { TF_NER_FINETUNED_MODELS = {
......
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