Merge branch 'main' into skylion007/fix-betas-bug

mosaicml · Mar 6, 2024 · 26d88bb · 26d88bb
2 parents 88f5518 + daddaef
commit 26d88bb
Show file tree

Hide file tree

Showing 5 changed files with 7 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,11 @@ compute:
 integrations:
 - integration_type: git_repo
   git_repo: mosaicml/examples
-  git_branch: v0.0.4
-  pip_install: .[stable-diffusion]
+  git_commit: fa3d60e71f1a9b10dc8b0c6b3933d55849234def
   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,12 @@ compute:
 integrations:
 - integration_type: git_repo
   git_repo: mosaicml/examples
-  pip_install: .[stable-diffusion-dreambooth]
+  git_commit: fa3d60e71f1a9b10dc8b0c6b3933d55849234def
   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