From b8dcea9e03a9224f8fd90bc14f231a3c39d4be16 Mon Sep 17 00:00:00 2001 From: austin Date: Wed, 31 Jan 2024 19:21:31 -0800 Subject: [PATCH] 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 | 2 ++ .../stable_diffusion_dreambooth/requirements.txt | 1 + .../stable_diffusion_dreambooth/yamls/mcloud_run.yaml | 2 +- 5 files changed, 6 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..88770651a 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..e4b5a9fdd 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,5 @@ 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..5be35da76 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