lightonai
/

RITA_s

Text Generation

Model card Files Files and versions Community

DanielHesslow commited on Apr 28, 2022

Commit

e283837

•

1 Parent(s): 7f52a15

add model

Files changed (1) hide show

config.json +6 -3

config.json CHANGED Viewed

@@ -1,16 +1,19 @@
 {
-  "_name_or_path": "./test_model",
   "architectures": [
     "RITAModel"
   ],
   "auto_map": {
     "AutoConfig": "rita_configuration.RITAConfig",
-    "AutoModel": "rita_modeling.RITAModel"
   },
   "bos_token_id": [
     [
       [
-        50256
       ]
     ]
   ],

 {
+  "_name_or_path": "Seledorn/RITA_s",
   "architectures": [
     "RITAModel"
   ],
   "auto_map": {
     "AutoConfig": "rita_configuration.RITAConfig",
+    "AutoModel": "rita_modeling.RITAModel",
+    "AutoModelForCausalLM": "rita_modeling.RITAModel"
   },
   "bos_token_id": [
     [
       [
+        [
+          50256
+        ]
       ]
     ]
   ],