We recommend starting with a single host first and then moving to multihost.
Local development is a convenient way to run MaxDiffusion on a single host. It doesn't scale to multiple hosts.
- Create and SSH to a single-host TPU (v4-8).
- Clone MaxDiffusion in your TPU VM.
- Within the root directory of the MaxDiffusion
git
repo, install dependencies by running:
bash setup.sh MODE=stable
[GKE, recommended] Running MaxDiffusion with xpk - Quick Experimentation and Production support