My Conda environment was set up as follows:

git clone https://github.com/apple/ml-stable-diffusion.git

conda create -n python_playground python=3.10 -y

conda activate python_playground

cd ml-stable-diffusion

pip install -e .

pip install omegaconf

pip install pytorch-lightning

pip install safetensors

pip install --upgrade --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu

Maybe another package or two was installed because I was prompted to in an error message at some point.