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

..

model_cache

don't use cpu state_dict for model unpatching when executing on cpu (#6631)

2024-07-18 15:34:01 -04:00

model_loaders

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

__init__.py

Apply ruff rule to disallow all relative imports.

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

load_base.py

Load single-file checkpoints directly without conversion (#6510)

2024-06-27 17:31:28 -04:00

load_default.py

Load single-file checkpoints directly without conversion (#6510)

2024-06-27 17:31:28 -04:00

memory_snapshot.py

Add ruff rule to disallow relative parent imports.

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

model_loader_registry.py

Apply ruff rule to disallow all relative imports.

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

model_util.py

Run ruff, setup initial text to image node

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

optimizations.py

final tidying before marking PR as ready for review

2024-03-01 10:42:33 +11:00