From fa3d60e71f1a9b10dc8b0c6b3933d55849234def Mon Sep 17 00:00:00 2001 From: Austin Date: Mon, 5 Feb 2024 18:05:53 -0800 Subject: [PATCH] update mcloud yamls + requirements for sd and sd-dreambooth (#446) * update mcloud yamls + requirements for sd and sd-dreambooth --- examples/end-to-end-examples/stable_diffusion/requirements.txt | 1 + .../end-to-end-examples/stable_diffusion/yamls/mcloud_run.yaml | 3 +-- .../stable_diffusion_dreambooth/requirements-cpu.txt | 1 + .../stable_diffusion_dreambooth/requirements.txt | 1 + .../stable_diffusion_dreambooth/yamls/mcloud_run.yaml | 2 +- 5 files changed, 5 insertions(+), 3 deletions(-) diff --git a/examples/end-to-end-examples/stable_diffusion/requirements.txt b/examples/end-to-end-examples/stable_diffusion/requirements.txt index 4e0acbcd7..c7f9260b7 100644 --- a/examples/end-to-end-examples/stable_diffusion/requirements.txt +++ b/examples/end-to-end-examples/stable_diffusion/requirements.txt @@ -10,6 +10,7 @@ omegaconf==2.3.0 # -------- diffusion -------- # diffusers[torch]==0.11.1 transformers[torch]==4.25.1 +datasets==2.14.6 pydantic==1.10.10 scipy==1.11.1 diff --git a/examples/end-to-end-examples/stable_diffusion/yamls/mcloud_run.yaml b/examples/end-to-end-examples/stable_diffusion/yamls/mcloud_run.yaml index b88efd3c6..16548198e 100644 --- a/examples/end-to-end-examples/stable_diffusion/yamls/mcloud_run.yaml +++ b/examples/end-to-end-examples/stable_diffusion/yamls/mcloud_run.yaml @@ -10,11 +10,10 @@ compute: integrations: - integration_type: git_repo git_repo: mosaicml/examples - git_branch: v0.0.4 - pip_install: .[stable-diffusion] ssh_clone: false command: | cd examples/examples/stable_diffusion + pip install -r requirements.txt composer main.py /mnt/config/parameters.yaml parameters: diff --git a/examples/end-to-end-examples/stable_diffusion_dreambooth/requirements-cpu.txt b/examples/end-to-end-examples/stable_diffusion_dreambooth/requirements-cpu.txt index 0adbed3b6..df2960da5 100644 --- a/examples/end-to-end-examples/stable_diffusion_dreambooth/requirements-cpu.txt +++ b/examples/end-to-end-examples/stable_diffusion_dreambooth/requirements-cpu.txt @@ -9,3 +9,4 @@ omegaconf # -------- diffusion -------- # diffusers[torch]==0.11.1 transformers[torch]==4.25.1 +datasets==2.14.6 diff --git a/examples/end-to-end-examples/stable_diffusion_dreambooth/requirements.txt b/examples/end-to-end-examples/stable_diffusion_dreambooth/requirements.txt index 2666d6297..dc1334655 100644 --- a/examples/end-to-end-examples/stable_diffusion_dreambooth/requirements.txt +++ b/examples/end-to-end-examples/stable_diffusion_dreambooth/requirements.txt @@ -9,6 +9,7 @@ omegaconf # -------- diffusion -------- # diffusers[torch]==0.11.1 transformers[torch]==4.25.1 +datasets==2.14.6 # -------- xformers -------- # xformers diff --git a/examples/end-to-end-examples/stable_diffusion_dreambooth/yamls/mcloud_run.yaml b/examples/end-to-end-examples/stable_diffusion_dreambooth/yamls/mcloud_run.yaml index a218f4d1d..6cb102346 100644 --- a/examples/end-to-end-examples/stable_diffusion_dreambooth/yamls/mcloud_run.yaml +++ b/examples/end-to-end-examples/stable_diffusion_dreambooth/yamls/mcloud_run.yaml @@ -10,11 +10,11 @@ compute: integrations: - integration_type: git_repo git_repo: mosaicml/examples - pip_install: .[stable-diffusion-dreambooth] ssh_clone: false command: | cd examples/examples/stable_diffusion_dreambooth + pip install -r requirements.txt composer main.py /mnt/config/parameters.yaml parameters: # Name of the training run used for checkpointing and other logging