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

Our first Diffusion model: FLUX #758

Merged
merged 54 commits into from
Sep 24, 2024
Merged

Our first Diffusion model: FLUX #758

merged 54 commits into from
Sep 24, 2024

Conversation

EricLBuehler
Copy link
Owner

black-forest-labs/FLUX.1-schnell is a 12 billion parameter rectified flow transformer capable of generating images from text descriptions. Uses time step distillation (fewer sampling steps, but max embed seq len of 256).

black-forest-labs/FLUX.1-dev is a 12 billion parameter rectified flow transformer capable of generating images from text descriptions. Uses guidance distillation (no limitations on max embed seq len).

Details: https://huggingface.co/docs/diffusers/en/api/pipelines/flux#flux

@EricLBuehler EricLBuehler added new feature New feature or request models Additions to model or architectures labels Sep 7, 2024
Copy link

github-actions bot commented Sep 7, 2024

Code Metrics Report
  ===============================================================================
 Language            Files        Lines         Code     Comments       Blanks
===============================================================================
 C Header                2           35           28            0            7
 Dockerfile              1           34           25            0            9
 Happy                   1          442          369            0           73
 JSON                   12          105          104            0            1
 Python                 48         2136         1816           64          256
 TOML                   20          615          551            2           62
 YAML                    2           21           19            2            0
-------------------------------------------------------------------------------
 Jupyter Notebooks       4            0            0            0            0
 |- Markdown             2           77           32           31           14
 |- Python               2          196          169            1           26
 (Total)                            273          201           32           40
-------------------------------------------------------------------------------
 Markdown               32         2323            0         1772          551
 |- BASH                 5          101           98            0            3
 |- JSON                 1           12           12            0            0
 |- Python               5           92           82            0           10
 |- Rust                 7          444          398           22           24
 |- TOML                 2           75           63            0           12
 (Total)                           3047          653         1794          600
-------------------------------------------------------------------------------
 Rust                  237        67220        60721         1315         5184
 |- Markdown           115         1140           25         1048           67
 (Total)                          68360        60746         2363         5251
===============================================================================
 Total                 360        72931        63633         3155         6143
===============================================================================
  

@EricLBuehler
Copy link
Owner Author

RUST_BACKTRACE=1 cargo run --features cuda -- --port 1234 diffusion-plain -m black-forest-labs/FLUX.1-schnell -a flux --dtype f16

@EricLBuehler
Copy link
Owner Author

RUST_BACKTRACE=1 cargo run --features cuda -- --port 1234 diffusion-plain -m black-forest-labs/FLUX.1-schnell -a flux

@EricLBuehler EricLBuehler merged commit ff2ccd5 into master Sep 24, 2024
12 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
models Additions to model or architectures new feature New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant