Edvards
/

my-awesome-mnist_cnn

model_hub_mixin

pytorch_model_hub_mixin

Inference Endpoints

Model card Files Files and versions Community

Edvards commited on Jun 20

Commit

f89ed15

•

1 Parent(s): 093b9b3

Upload model

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -2,7 +2,7 @@
   "architectures": [
     "MyModel"
   ],
-  "model_type": "custom-resnet-tutorial",
   "torch_dtype": "float32",
   "transformers_version": "4.41.2"
 }

   "architectures": [
     "MyModel"
   ],
+  "model_type": "resnet",
   "torch_dtype": "float32",
   "transformers_version": "4.41.2"
 }