Update config_codesage.py
Browse files- config_codesage.py +6 -0
config_codesage.py
CHANGED
|
@@ -4,6 +4,12 @@
|
|
| 4 |
|
| 5 |
from transformers.configuration_utils import PretrainedConfig
|
| 6 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 7 |
|
| 8 |
class CodeSageConfig(PretrainedConfig):
|
| 9 |
model_type = "codesage"
|
|
|
|
| 4 |
|
| 5 |
from transformers.configuration_utils import PretrainedConfig
|
| 6 |
|
| 7 |
+
CODESAGE_PRETRAINED_CONFIG_ARCHIVE_MAP = {
|
| 8 |
+
"codesage/codesage-small": "https://huggingface.co/codesage/codesage-small/resolve/main/config.json",
|
| 9 |
+
"codesage/codesage-base": "https://huggingface.co/codesage/codesage-base/resolve/main/config.json",
|
| 10 |
+
"codesage/codesage-large": "https://huggingface.co/codesage/codesage-large/resolve/main/config.json",
|
| 11 |
+
}
|
| 12 |
+
|
| 13 |
|
| 14 |
class CodeSageConfig(PretrainedConfig):
|
| 15 |
model_type = "codesage"
|