Skip to content

Commit

Permalink
update mcloud yamls + requirements for sd and sd-dreambooth (#446)
Browse files Browse the repository at this point in the history
* update mcloud yamls + requirements for sd and sd-dreambooth
  • Loading branch information
A-Jacobson authored Feb 6, 2024
1 parent 7003793 commit fa3d60e
Show file tree
Hide file tree
Showing 5 changed files with 5 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -9,3 +9,4 @@ omegaconf
# -------- diffusion -------- #
diffusers[torch]==0.11.1
transformers[torch]==4.25.1
datasets==2.14.6
Original file line number Diff line number Diff line change
Expand Up @@ -9,6 +9,7 @@ omegaconf
# -------- diffusion -------- #
diffusers[torch]==0.11.1
transformers[torch]==4.25.1
datasets==2.14.6

# -------- xformers -------- #
xformers
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit fa3d60e

Please sign in to comment.