First change into the JaxDiffusion
directory and follow the instructions below to set up the repository:
-
Install
uv
:curl -LsSf https://astral.sh/uv/install.sh | sh
-
Install Python 3.11 using
uv
:uv python install 3.11
-
Create a virtual environment with Python 3.11 and sync:
uv venv --python 3.11 source .venv/bin/activate uv sync