Update model weights - ensure safetensors tag

config.json

@@ -1,25 +1,27 @@
 {
-  "model_type": "bt4",
-  "architectures": ["BT4"],
-  "embedding_size": 1024,
+  "architectures": [
+    "BT4"
+  ],
+  "auto_map": {
+    "AutoConfig": "model.BT4Config",
+    "AutoModel": "model.BT4"
+  },
+  "dropout_rate": 0.0,
+  "dtype": "float32",
   "embedding_dense_sz": 512,
-  "encoder_layers": 15,
+  "embedding_size": 1024,
   "encoder_d_model": 1024,
-  "encoder_heads": 32,
   "encoder_dff": 1536,
-  "dropout_rate": 0.0,
+  "encoder_heads": 32,
+  "encoder_layers": 15,
+  "model_type": "bt4",
   "pol_embedding_size": 1024,
   "policy_d_model": 1024,
-  "val_embedding_size": 128,
-  "use_smolgen": true,
+  "smol_activation": "swish",
+  "smol_gen_sz": 256,
   "smol_hidden_channels": 32,
   "smol_hidden_sz": 256,
-  "smol_gen_sz": 256,
-  "smol_activation": "swish",
-  "auto_map": {
-    "AutoConfig": "model.BT4Config",
-    "AutoModel": "model.BT4"
-  },
-  "torch_dtype": "float32"
+  "transformers_version": "4.57.1",
+  "use_smolgen": true,
+  "val_embedding_size": 128
 }
-

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f32a089786a4bd29525016b14add9bb42c601accf1c813a3a40b2f157b2fc04e
-size 835207344
+oid sha256:59ccc54f97ee20a6158a1947fae0f7143f2a27c611537d10a703960eee4f5062
+size 835207368