"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "3996041d0ae23ce23dfb8a343e6344f2f8d54c16"
Commit 906581ae authored by VictorSanh's avatar VictorSanh
Browse files

add s3 links for dilbert (+fix small typo)

parent b247b0d8
...@@ -37,11 +37,11 @@ logger = logging.getLogger(__name__) ...@@ -37,11 +37,11 @@ logger = logging.getLogger(__name__)
DILBERT_PRETRAINED_MODEL_ARCHIVE_MAP = { DILBERT_PRETRAINED_MODEL_ARCHIVE_MAP = {
'dilbert-base-uncased': None, # TODO(Victor) 'dilbert-base-uncased': "https://s3.amazonaws.com/models.huggingface.co/bert/dilbert-base-uncased-pytorch_model.bin"
} }
DILBERT_PRETRAINED_CONFIG_ARCHIVE_MAP = { DILBERT_PRETRAINED_CONFIG_ARCHIVE_MAP = {
'dilbert-base-uncased': None, #TODO(Victor) 'dilbert-base-uncased': "https://s3.amazonaws.com/models.huggingface.co/bert/dilbert-base-uncased-config.json"
} }
...@@ -85,7 +85,7 @@ class DilBertConfig(PretrainedConfig): ...@@ -85,7 +85,7 @@ class DilBertConfig(PretrainedConfig):
self.tie_weights_ = tie_weights_ self.tie_weights_ = tie_weights_
else: else:
raise ValueError("First argument must be either a vocabulary size (int)" raise ValueError("First argument must be either a vocabulary size (int)"
"or the path to a pretrained model config file (str)") " or the path to a pretrained model config file (str)")
### UTILS AND BUILDING BLOCKS OF THE ARCHITECTURE ### ### UTILS AND BUILDING BLOCKS OF THE ARCHITECTURE ###
......
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