Ryan Dick 0c5e11f521 Fix FLUX output image clamping. And a few other minor fixes to make inference work with the full bfloat16 FLUX transformer model.

2024-08-26 20:17:50 -04:00

..

load

Fix FLUX output image clamping. And a few other minor fixes to make inference work with the full bfloat16 FLUX transformer model.

2024-08-26 20:17:50 -04:00

metadata

Apply ruff rule to disallow all relative imports.

2024-07-04 09:35:37 -04:00

util

Install sub directories with folders correctly, ensure consistent dtype of tensors in flux pipeline and vae

2024-08-26 20:17:50 -04:00

__init__.py

Apply ruff rule to disallow all relative imports.

2024-07-04 09:35:37 -04:00

config.py

Add nf4 bnb quantized format

2024-08-26 20:17:50 -04:00

merge.py

Apply ruff rule to disallow all relative imports.

2024-07-04 09:35:37 -04:00

probe.py

Select dev/schnell based on state dict, use correct max seq len based on dev/schnell, and shift in inference, separate vae flux params into separate config

2024-08-26 20:17:50 -04:00

search.py

docs(mm): update ModelSearch

2024-03-28 12:35:41 +11:00

starter_models.py

feat(ui): update qr code cnet starter model

2024-07-26 13:34:32 +10:00