update: autoclass

config.json/config.json

@@ -1,6 +1,6 @@
 {
   "input_dim": 4,
-  "model_type": "test_model",
+  "model_type": "copperlight/test-model",
   "output_dim": 16,
   "transformers_version": "4.54.1"
 }

model.py

@@ -2,7 +2,7 @@ from transformers import PreTrainedModel, PretrainedConfig
 from torch import nn
 
 class TestConfig(PretrainedConfig):
-    model_type = "test_model" 
+    model_type = "copperlight/test-model" 
     
     def __init__(self, input_dim=4, output_dim=16, **kwargs):
         super().__init__(**kwargs)

pytorch_model.bin/config.json

@@ -3,7 +3,7 @@
     "TestModel"
   ],
   "input_dim": 4,
-  "model_type": "test_model",
+  "model_type": "copperlight/test-model",
   "output_dim": 16,
   "torch_dtype": "float32",
   "transformers_version": "4.54.1"

pytorch_model.bin/model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:4d309df6d7ad88e731d193ac5329b87ea597b1175a8b0ba7b33ba05d15821fb7
+oid sha256:829c8d7855132961e13f97016f8d650b346b4945ba34d9680798c74c90de5fff
 size 496

test.ipynb

@@ -0,0 +1,19 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "8ce5e1b0",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "language_info": {
+   "name": "python"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}