Make public config safetensors-only

config.json

@@ -18,7 +18,6 @@
   ],
   "foundation_checkpoint": "weights/Final_V3-model_state.safetensors",
   "foundation_checkpoint_format": "safetensors",
-  "foundation_pytorch_model_state_checkpoint": "weights/Final_V3-model_state.pt",
   "full_training_checkpoints": {
     "location": "Wasabi object storage",
     "note": "Optimizer-bearing full training checkpoints are intentionally not part of the public Hugging Face payload; the public release contains model-state weights only."
@@ -48,6 +47,15 @@
     "unrestricted de novo molecule generation",
     "calibrated top-1 molecular identification without external validation"
   ],
+  "object_storage_pytorch_weights": {
+    "files": [
+      "weights/Final_V3-model_state.pt",
+      "weights/NexaMass-V3-Struct-model_state.pt"
+    ],
+    "location": "Wasabi object storage",
+    "note": "PyTorch .pt model-state files are retained in object storage. The public Hugging Face payload is safetensors-only.",
+    "path": "wasabi:nexa-ms/models/dreams_pretrain_v1/checkpoints/NexaMass-V3-Struct/hf_release/weights/"
+  },
   "outputs": {
     "morgan_fingerprint_logits": 2048,
     "retrieval_target_dim": 256,
@@ -57,7 +65,11 @@
   "parameter_count": 14106690,
   "primary_checkpoint": "weights/NexaMass-V3-Struct-model_state.safetensors",
   "primary_checkpoint_format": "safetensors",
-  "pytorch_model_state_checkpoint": "weights/NexaMass-V3-Struct-model_state.pt",
+  "public_weight_files": [
+    "weights/NexaMass-V3-Struct-model_state.safetensors",
+    "weights/Final_V3-model_state.safetensors"
+  ],
+  "public_weight_format": "safetensors",
   "recommended_inference": "Use the structure-aligned model-state checkpoint for spectrum embeddings and Morgan fingerprint probability vectors, then compare against a candidate molecular fingerprint bank.",
   "task": "MS/MS spectral representation learning and RDKit Morgan fingerprint alignment"
 }