Instructions to use codesage/codesage-small with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use codesage/codesage-small with Transformers:
# Load model directly from transformers import CodeSage model = CodeSage.from_pretrained("codesage/codesage-small", trust_remote_code=True, dtype="auto") - Notebooks
- Google Colab
- Kaggle
Update config_codesage.py
Browse files- config_codesage.py +1 -1
config_codesage.py
CHANGED
|
@@ -6,7 +6,7 @@ from transformers.configuration_utils import PretrainedConfig
|
|
| 6 |
|
| 7 |
|
| 8 |
class CodeSageConfig(PretrainedConfig):
|
| 9 |
-
model_type = "
|
| 10 |
|
| 11 |
def __init__(
|
| 12 |
self,
|
|
|
|
| 6 |
|
| 7 |
|
| 8 |
class CodeSageConfig(PretrainedConfig):
|
| 9 |
+
model_type = "codesage"
|
| 10 |
|
| 11 |
def __init__(
|
| 12 |
self,
|