From e522c18a375106dbf18b43f38736184afbddd2a2 Mon Sep 17 00:00:00 2001 From: Juan Acevedo Date: Wed, 4 Dec 2024 21:53:31 +0000 Subject: [PATCH] update setup instructions. --- README.md | 2 +- docs/getting_started/first_run.md | 4 +--- setup.sh | 3 +++ 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 54fe093..8f21ff5 100644 --- 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 index cabc720..900c3cf 100644 --- 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 index a8f32fd..5f13f72 100644 --- 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 . \ No newline at end of file