R0CKSTAR 7a72f34ce4 Add supports_float64() to OmniPlatform and clean up MPS (#2488) ...

Signed-off-by: Xiaodong Ye <yeahdongcn@gmail.com>

2026-04-07 11:15:02 +08:00

..

__init__.py

[model] support FLUX.1-Kontext-dev (#561)

2026-03-23 17:17:58 +00:00

flux_pipeline_mixin.py

[model] support FLUX.1-Kontext-dev (#561)

2026-03-23 17:17:58 +00:00

flux_transformer.py

[Feat] support for multi-block layerwise offloading, fix top-level parameters/buffers staying on CPU (#1486)

2026-04-06 11:49:35 +08:00

pipeline_flux_kontext.py

Add supports_float64() to OmniPlatform and clean up MPS (#2488)

2026-04-07 11:15:02 +08:00

pipeline_flux.py

[Core] Unified quantization framework (#1764)

2026-03-24 18:56:55 +08:00