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.

Sign up for GitHub

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

Jump to bottom

Add Shuttle 3 Diffusion image generator #77

Merged
merged 7 commits into from
Jan 20, 2025
Merged

Add Shuttle 3 Diffusion image generator #77

merged 7 commits into from
Jan 20, 2025

Conversation

sokovninn
Copy link
Member

@sokovninn sokovninn commented Jan 20, 2025 •
edited
Loading

This PR introduces the Shuttle 3 Diffusion image generator. To ensure compatibility with GPUs with less than 24GB VRAM, it is quantized. For a 512x512 image size, its speed is comparable to SDXL-Turbo.

Examples:

image_7_a016ce71fa5a488191c74d5239413d69
image_4_92d85ecae4e24eb087812f994f6d777e
image_5_91e00968c1ce4ce8afe65aed2bc5ee0f

@sokovninn sokovninn self-assigned this Jan 20, 2025
@github-actions GitHub Actions
Copy link

github-actions bot commented Jan 20, 2025 •
edited
Loading

☂️ Python Coverage

current status: ✅

Overall Coverage

Lines Covered Coverage Threshold Status
1790 1109 62% 0% 🟢

New Files

File Coverage Status
datadreamer/image_generation/shuttle_3_image_generator.py 24% 🟢
TOTAL 24% 🟢

Modified Files

File Coverage Status
datadreamer/image_generation/init.py 100% 🟢
datadreamer/pipelines/generate_dataset_from_scratch.py 38% 🟢
datadreamer/utils/config.py 100% 🟢
TOTAL 79% 🟢

updated for commit: 76e37e7 by action🐍

HonzaCuhel
HonzaCuhel approved these changes Jan 20, 2025
View reviewed changes
Copy link
Contributor

@HonzaCuhel HonzaCuhel left a comment •
edited
Loading

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Left a small comment, but otherwise LGTM! The images look fantastic!

examples/generate_instance_segmentation_dataset_and_train_yolo.ipynb
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Missing the shuttle-3 mention, also in generate_dataset_and_train_yolo.ipynb.

@sokovninn sokovninn merged commit 1aee13a into dev Jan 20, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@HonzaCuhel HonzaCuhel HonzaCuhel approved these changes

@klemen1999 klemen1999 Awaiting requested review from klemen1999

@tersekmatija tersekmatija Awaiting requested review from tersekmatija

Assignees

@sokovninn sokovninn

Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@sokovninn @HonzaCuhel @actions-user