Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

MLCube implementation for Stable Diffusion #696

Open
wants to merge 6 commits into
base: master
Choose a base branch
from

Conversation

davidjurado
Copy link
Contributor

@davidjurado davidjurado commented Jan 16, 2024

MLCube for Stable Diffusion

MLCube™ GitHub repository. MLCube™ wiki.

Project setup

An important requirement is that you must have Docker installed.

# Create Python environment and install MLCube Docker runner 
virtualenv -p python3 ./env && source ./env/bin/activate && pip install pip==24.0
pip install mlcube-docker
# Fetch the implementation from GitHub
git clone https://github.com/mlcommons/training && cd ./training
git fetch origin pull/696/head:feature/mlcube_sd && git checkout feature/mlcube_sd
cd ./stable_diffusion/mlcube

Inside the mlcube directory run the following command to check implemented tasks.

mlcube describe

MLCube tasks

  • Core tasks:

Download dataset.

mlcube run --task=download_data

Download models.

mlcube run --task=download_models

Train.

mlcube run --task=train
  • Demo tasks:

Here is a video explaining the demo steps:

IMAGE ALT TEXT HERE

Download demo dataset.

mlcube run --task=download_demo

Download models.

mlcube run --task=download_models

Train demo.

mlcube run --task=demo

Execute the complete pipeline

You can execute the complete pipeline with one single command.

  • Core pipeline:
mlcube run --task=download_data,download_models,train
  • Demo pipeline:

Tested in an Nvidia A100 (40G)

mlcube run --task=download_demo,download_models,demo

Note: To rebuild the image use the flag: -Pdocker.build_strategy=always during the mlcube run command.

@davidjurado davidjurado requested a review from a team as a code owner January 16, 2024 16:24
Copy link

github-actions bot commented Jan 16, 2024

MLCommons CLA bot All contributors have signed the MLCommons CLA ✍️ ✅

@davidjurado davidjurado changed the title [WIP] Add MLCube implementation for Stable Diffusion Add MLCube implementation for Stable Diffusion Feb 22, 2024
@davidjurado davidjurado changed the title Add MLCube implementation for Stable Diffusion MLCube implementation for Stable Diffusion Jul 31, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant