update

2026-05-14 18:37:18 +08:00
parent 5e6d8046e1
commit 7a2ffc91a2
15 changed files with 200 additions and 56 deletions
@@ -76,6 +76,11 @@ def get_inpaint_predictor(model_name: str):
        _inpaint_cache[model_name] = _copy
        return _copy

+    if model_name == "flux_fill":
+        pred, _ = build_inpaint_predictor(UnifiedInpaintConfig(backend=InpaintBackend.FLUX_FILL))
+        _inpaint_cache[model_name] = pred
+        return pred
+
    if model_name == "sdxl_inpaint":
        pred, _ = build_inpaint_predictor(UnifiedInpaintConfig(backend=InpaintBackend.SDXL_INPAINT))
        _inpaint_cache[model_name] = pred