Upload SAGViTClassifier

config.json

@@ -1,17 +1,23 @@
-{
-  "model_type": "sagvit",
-  "d_model": 64,
-  "dim_feedforward": 64,
-  "gcn_hidden": 128,
-  "gcn_out": 64,
-  "hidden_mlp_features": 64,
-  "in_channels": 2560,
-  "nhead": 4,
-  "num_classes": 10,
-  "num_layers": 2,
-  "patch_size": [
-    4,
-    4
-  ],
-  "use_safetensors": true
-}
+{
+  "_name_or_path": "shravvvv/SAG-ViT",
+  "architectures": [
+    "SAGViTClassifier"
+  ],
+  "d_model": 64,
+  "dim_feedforward": 64,
+  "gcn_hidden": 128,
+  "gcn_out": 64,
+  "hidden_mlp_features": 64,
+  "in_channels": 2560,
+  "model_type": "sagvit",
+  "nhead": 4,
+  "num_classes": 10,
+  "num_layers": 2,
+  "patch_size": [
+    4,
+    4
+  ],
+  "torch_dtype": "float32",
+  "transformers_version": "4.47.0",
+  "use_safetensors": true
+}

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:1d51dccc77d313f786884f38c6d212e9c06c207938ea748755632c9a3c60e6bb
-size 27009168
+oid sha256:d53188f21736a64e352e9e2c3d73cc6e017177f19cae7de9d5f4b5ee282292ef
+size 32389992