update mcloud yamls + requirements for sd and sd-dreambooth (#446)

* update mcloud yamls + requirements for sd and sd-dreambooth
mosaicml · Feb 6, 2024 · fa3d60e · fa3d60e
1 parent 7003793
commit fa3d60e
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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