update setup instructions.

AI-Hypercomputer · Dec 4, 2024 · e522c18 · e522c18
1 parent a58a75f
commit e522c18
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -188,7 +188,7 @@ MaxDiffusion started as a fork of [Diffusers](https://github.com/huggingface/dif
 
 Whether you are forking MaxDiffusion for your own needs or intending to contribute back to the community, a full suite of tests can be found in `tests` and `src/maxdiffusion/tests`.
 
-To run unit tests, you'll need to [install gcsfuse](https://cloud.google.com/storage/docs/cloud-storage-fuse/install) then simply run:
+To run unit tests simply run:
 ```
 python -m pytest
 ```

diff --git a/docs/getting_started/first_run.md b/docs/getting_started/first_run.md
@@ -12,9 +12,7 @@ multiple hosts.
 1. Clone MaxDiffusion in your TPU VM.
 1. Within the root directory of the MaxDiffusion `git` repo, install dependencies by running:
 ```bash
-pip3 install jax[tpu] -f https://storage.googleapis.com/jax-releases/libtpu_releases.html
-pip3 install -r requirements.txt
-pip3 install .
+bash setup.sh MODE=stable
 ```
 
 ## Getting Starting: Multihost development

diff --git a/setup.sh b/setup.sh
@@ -108,3 +108,6 @@ fi
 
 # Install dependencies from requirements.txt
 pip3 install -U -r requirements.txt
+
+# Install maxdiffusion
+pip3 install -e -U .