KKNakka/MammoSAE

This is the first SAE model trained on a Mammo-CLIP breast foundation model.

Model Information

This release contains single SAE. After cloning our MammoSAE repo, you can load them with the following snippet:

from huggingface_hub import hf_hub_download

ckpt_path = hf_hub_download(
      repo_id=f"KKNakka/MammoSAE",
      filename=f"sparse_autoencoder_final.pt",
      repo_type="model"
  )

print(f"Loading SAE checkpoint from: {ckpt_path}")
state_dict = torch.load(ckpt_path)
autoencoder.load_state_dict(state_dict)