This repository contains code accompanying 'Schrödinger’s Bat: Diffusion Models Sometimes Generate Polysemous Words in Superposition' by Jennifer C. White and Ryan Cotterell (pre-print).

The notebook experiments.ipynb contains code that will run all of the experiments described in the paper.

An authorisation key for Stable Diffusion is required.

We ran the original experiments using Google Colab.

@misc{white-cotterell-2022-diffusion,
    title = "Schr\"{o}dinger's Bat: Diffusion Models Sometimes Generate Polysemous Words in Superposition",
    author = "White, Jennifer C.  and
      Cotterell, Ryan",
    year = "2022",
    url = {https://arxiv.org/pdf/2211.13095.pdf}.
}