Unverified Commit 8f8b3cbc authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Fix checkpoint name (#17484)


Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 400b3093
...@@ -21,7 +21,7 @@ from ...utils import logging ...@@ -21,7 +21,7 @@ from ...utils import logging
logger = logging.get_logger(__name__) logger = logging.get_logger(__name__)
GPT_NEOX_PRETRAINED_CONFIG_ARCHIVE_MAP = { GPT_NEOX_PRETRAINED_CONFIG_ARCHIVE_MAP = {
"eleutherai/gpt-neox-20b": "https://huggingface.co/eleutherai/gpt-neox-20b/resolve/main/config.json", "EleutherAI/gpt-neox-20b": "https://huggingface.co/EleutherAI/gpt-neox-20b/resolve/main/config.json",
# See all GPTNeoX models at https://huggingface.co/models?filter=gpt_neox # See all GPTNeoX models at https://huggingface.co/models?filter=gpt_neox
} }
...@@ -31,7 +31,7 @@ class GPTNeoXConfig(PretrainedConfig): ...@@ -31,7 +31,7 @@ class GPTNeoXConfig(PretrainedConfig):
This is the configuration class to store the configuration of a [`GPTNeoXModel`]. It is used to instantiate an This is the configuration class to store the configuration of a [`GPTNeoXModel`]. It is used to instantiate an
GPTNeoX model according to the specified arguments, defining the model architecture. Instantiating a configuration GPTNeoX model according to the specified arguments, defining the model architecture. Instantiating a configuration
with the defaults will yield a similar configuration to that of the GPTNeoX with the defaults will yield a similar configuration to that of the GPTNeoX
[eleutherai/gpt-neox-20b](https://huggingface.co/eleutherai/gpt-neox-20b) architecture. [EleutherAI/gpt-neox-20b](https://huggingface.co/EleutherAI/gpt-neox-20b) architecture.
Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the
documentation from [`PretrainedConfig`] for more information. documentation from [`PretrainedConfig`] for more information.
......
...@@ -32,7 +32,7 @@ VOCAB_FILES_NAMES = {"vocab_file": "vocab.json", "merges_file": "merges.txt", "t ...@@ -32,7 +32,7 @@ VOCAB_FILES_NAMES = {"vocab_file": "vocab.json", "merges_file": "merges.txt", "t
PRETRAINED_VOCAB_FILES_MAP = { PRETRAINED_VOCAB_FILES_MAP = {
"tokenizer_file": { "tokenizer_file": {
"eleutherai/gpt-neox-20b": "https://huggingface.co/EleutherAI/gpt-neox-20b/resolve/main/tokenizer.json", "EleutherAI/gpt-neox-20b": "https://huggingface.co/EleutherAI/gpt-neox-20b/resolve/main/tokenizer.json",
}, },
} }
......
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