Vitalis_LFM2.5_Cortex.GGUF / src /brain /checkpoint_manager.py
FerrellSyntheticIntelligence's picture
Upload folder using huggingface_hub
d2a5f5a verified
Raw
History Blame Contribute Delete
224 Bytes
import torch
import os
def save_vitalis_weights(model, path="checkpoints/vitalis_v1.pt"):
os.makedirs(os.path.dirname(path), exist_ok=True)
torch.save(model.state_dict(), path)
print(f"Weights saved to {path}")