SageLite
/

SageLite-s

Model card Files Files and versions

Dejiao Z commited on Dec 5, 2024

Commit

812f107

·

1 Parent(s): 6172c4e

fixed config error

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -6,7 +6,7 @@
     "auto_map": {
         "AutoConfig": "config_sagelite.SageLiteConfig",
         "AutoTokenizer": "tokenization_sagelite.SageLiteTokenizer",
-        "AutoModel": "modeling_codesage.SageLiteModel",
         "AutoModelForMaskedLM": "modeling_sagelite.SageLiteForMaskedLM",
         "AutoModelForSequenceClassification": "modeling_sagelite.SageLiteForSequenceClassification"
     },

     "auto_map": {
         "AutoConfig": "config_sagelite.SageLiteConfig",
         "AutoTokenizer": "tokenization_sagelite.SageLiteTokenizer",
+        "AutoModel": "modeling_sagelite.SageLiteModel",
         "AutoModelForMaskedLM": "modeling_sagelite.SageLiteForMaskedLM",
         "AutoModelForSequenceClassification": "modeling_sagelite.SageLiteForSequenceClassification"
     },