Skip to content

Commit 8b52e8b

Browse files
committed
Fixing broken compatibilities
1 parent 4b4b9c8 commit 8b52e8b

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

wildedge/client.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,12 @@ class WildEdge:
9393
"tensorflow",
9494
"huggingface",
9595
}
96+
PATCH_INSTALLERS = {
97+
"gguf": GgufExtractor.install_auto_load_patch,
98+
"onnx": OnnxExtractor.install_auto_load_patch,
99+
"timm": PytorchExtractor.install_timm_patch,
100+
"tensorflow": TensorflowExtractor.install_auto_load_patch,
101+
}
96102

97103
def __init__(
98104
self,
@@ -243,12 +249,6 @@ def register_model(
243249

244250
return handle
245251

246-
PATCH_INSTALLERS = {
247-
"gguf": GgufExtractor.install_auto_load_patch,
248-
"onnx": OnnxExtractor.install_auto_load_patch,
249-
"timm": PytorchExtractor.install_timm_patch,
250-
"tensorflow": TensorflowExtractor.install_auto_load_patch,
251-
}
252252
def _find_extractor(self, model_obj: object) -> BaseExtractor | None:
253253
for candidate in DEFAULT_EXTRACTORS:
254254
if candidate.can_handle(model_obj):

0 commit comments

Comments
 (0)