malaysia-ai
/

malaysian-sfw-classifier

Text Classification

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

huseinzol05 commited on Jun 3, 2024

Commit

e46e4a7

·

verified ·

1 Parent(s): 6d8eb01

Upload MistralForSequenceClassification

Files changed (2) hide show

config.json +4 -5
model.safetensors +2 -2

config.json CHANGED Viewed

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "sfw-mistral-mlm",
   "architectures": [
     "MistralForSequenceClassification"
   ],
@@ -45,7 +45,7 @@
   "sliding_window": 4096,
   "tie_word_embeddings": false,
   "torch_dtype": "float32",
-  "transformers_version": "4.40.1",
   "use_cache": true,
   "vocab": [
     "racist",
@@ -53,9 +53,8 @@
     "psychiatric or mental illness",
     "sexist",
     "harassment",
-    "informative",
-    "safe for work",
-    "hate"
   ],
   "vocab_size": 32000
 }

 {
+  "_name_or_path": "malaysian-sfw",
   "architectures": [
     "MistralForSequenceClassification"
   ],
   "sliding_window": 4096,
   "tie_word_embeddings": false,
   "torch_dtype": "float32",
+  "transformers_version": "4.41.2",
   "use_cache": true,
   "vocab": [
     "racist",
     "psychiatric or mental illness",
     "sexist",
     "harassment",
+    "porn",
+    "safe for work"
   ],
   "vocab_size": 32000
 }

model.safetensors CHANGED Viewed

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c00de7ca5b461ca6f6c185f7945bf86127dbc5ca8804f6050f978093bab6c317
-size 664677336

 version https://git-lfs.github.com/spec/v1
+oid sha256:2f11ad64311a40128cb176a5290f925d53100e661a427da5c8445d8c613e0d6f
+size 664674264