From 356505cab23248ca9282282cdcff9d3192f419ec Mon Sep 17 00:00:00 2001 From: CalamitousFelicitousness Date: Sun, 8 Feb 2026 22:59:45 +0000 Subject: [PATCH] refactor: remove ONNX face upscaler support from detailer - Remove codeformer, restoreformer, GFPGANv1.4, and GPEN-BFR ONNX model URLs from the predefined list - Remove the .fp16 ONNX restorer code path that bypassed detailer processing to run face restoration directly --- modules/postprocess/yolo.py | 10 ---------- 1 file changed, 10 deletions(-) diff --git a/modules/postprocess/yolo.py b/modules/postprocess/yolo.py index bb0fd4caf..61f5f5661 100644 --- a/modules/postprocess/yolo.py +++ b/modules/postprocess/yolo.py @@ -21,10 +21,6 @@ predefined = [ # 'https://huggingface.co/vladmandic/yolo-detailers/resolve/main/anzhc-eyes-seg.pt', 'https://huggingface.co/vladmandic/yolo-detailers/resolve/main/anzhc-face-1024-seg-8n.pt', 'https://huggingface.co/vladmandic/yolo-detailers/resolve/main/anzhc-head-seg-8n.pt', - 'https://huggingface.co/netrunner-exe/Face-Upscalers-onnx/resolve/main/codeformer.fp16.onnx', - 'https://huggingface.co/netrunner-exe/Face-Upscalers-onnx/resolve/main/restoreformer.fp16.onnx', - 'https://huggingface.co/netrunner-exe/Face-Upscalers-onnx/resolve/main/GFPGANv1.4.fp16.onnx', - 'https://huggingface.co/netrunner-exe/Face-Upscalers-onnx/resolve/main/GPEN-BFR-512.fp16.onnx', ] load_lock = threading.Lock() @@ -314,12 +310,6 @@ class YoloRestorer(Detailer): shared.log.warning(f'Detailer: model="{name}" not loaded') continue - if name.endswith('.fp16'): # run gfpgan or codeformer directly and skip detailer processing - from modules.postprocess import restorer - np_image = restorer.restore(np_image, name, model, p.detailer_strength) - image = Image.fromarray(np_image) - continue - if image is None: image = Image.fromarray(np_image) items = self.predict(model, image)