File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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 ):
You can’t perform that action at this time.
0 commit comments