Transformers
PyTorch
code
custom_code
Inference Endpoints
codesage commited on
Commit
7fc7745
1 Parent(s): b8f6718

Update config_codesage.py

Browse files
Files changed (1) hide show
  1. 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"