change args in load_model

supervisely-ecosystem · Dec 2, 2024 · 4e8af85 · 4e8af85
1 parent 2714b20
commit 4e8af85
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/supervisely_integration/serve/rtdetrv2.py b/supervisely_integration/serve/rtdetrv2.py
@@ -27,7 +27,7 @@ class RTDETRv2(sly.nn.inference.ObjectDetection):
     # TODO: may be do it auto?
 
     def load_model(
-        self, model_source: str, model_files: dict, model_info: dict, device: str, runtime: str
+        self, model_files: dict, model_info: dict, model_source: str, device: str, runtime: str
     ):
         config_path = f'{CONFIG_DIR}/{model_files["config"]}'
         checkpoint_path = model_files["checkpoint"]