loubb
/

aria-medium-base

Model card Files Files and versions

loua19 commited on Jun 5, 2025

Commit

13425d0

·

1 Parent(s): 840c6a3

fix config

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
   "architectures": [
-    "AriaForSequenceEmbeddings"
   ],
   "bos_token_id": 0,
   "eos_token_id": 1,
@@ -18,6 +18,6 @@
   "auto_map": {
     "AutoConfig": "configuration_aria.AriaConfig",
     "AutoModel": "modeling_aria.AriaModel",
-    "AutoModelForCausalLM": "modeling_aria.AriaForCausalLM"
   }
 }

 {
   "architectures": [
+    "AriaForSequenceEmbedding"
   ],
   "bos_token_id": 0,
   "eos_token_id": 1,
   "auto_map": {
     "AutoConfig": "configuration_aria.AriaConfig",
     "AutoModel": "modeling_aria.AriaModel",
+    "AutoModelForCausalLM": "modeling_aria.AriaForSequenceEmbedding"
   }
 }