From 383aaca1ebe3f68d43e9d5d2dec5b6106e397f17 Mon Sep 17 00:00:00 2001 From: Chenlei Hu Date: Thu, 8 Feb 2024 03:53:41 +0000 Subject: [PATCH] Improve model filtering (#114) --- .../scripts/legacy_preprocessors.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/extensions-builtin/forge_legacy_preprocessors/scripts/legacy_preprocessors.py b/extensions-builtin/forge_legacy_preprocessors/scripts/legacy_preprocessors.py index 3e8ade8f..a2d4f8be 100644 --- a/extensions-builtin/forge_legacy_preprocessors/scripts/legacy_preprocessors.py +++ b/extensions-builtin/forge_legacy_preprocessors/scripts/legacy_preprocessors.py @@ -108,7 +108,10 @@ class LegacyPreprocessor(Preprocessor): return result -for k, v in legacy_preprocessors.items(): - p = LegacyPreprocessor(v) - p.name = k +for name, data in legacy_preprocessors.items(): + p = LegacyPreprocessor(data) + p.name = name + # Invert should not match any particular model. + if "invert" in name: + p.model_filename_filters = [] add_supported_preprocessor(p)