Allanatrix's picture
Upload NexaMass V3 safetensors weights and release metadata
97ced61 verified

NexaMass Runtime

nexamass_encoder.py is a minimal PyTorch loader for the model-state files in this repository.

Recommended checkpoint for inference:

from runtime.nexamass_encoder import ModelConfig, load_nexamass_model_state

model = load_nexamass_model_state("weights/NexaMass-V3-Struct-model_state.pt", ModelConfig())  # PyTorch fallback
# Safetensors weights are also provided for safer tensor-only loading: weights/NexaMass-V3-Struct-model_state.safetensors

Use weights/NexaMass-V3-Struct-model_state.safetensors for the RDKit/Morgan-aligned model and weights/Final_V3-model_state.safetensors for the self-supervised foundation parent. PyTorch model-state .pt fallbacks are also included. Optimizer-bearing full training checkpoints remain in object storage and are intentionally excluded from this public HF payload.