From 6fea9f7b17e5cf1cf1afff61cef84ebe138df188 Mon Sep 17 00:00:00 2001 From: Grzegorz Klimaszewski <166530809+grzegorz-roboflow@users.noreply.github.com> Date: Mon, 13 Jan 2025 18:47:01 +0100 Subject: [PATCH] Fix OwlV2.__init__ --- inference/models/owlv2/owlv2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/inference/models/owlv2/owlv2.py b/inference/models/owlv2/owlv2.py index d8bbb172d9..ab9c6126df 100644 --- a/inference/models/owlv2/owlv2.py +++ b/inference/models/owlv2/owlv2.py @@ -293,8 +293,8 @@ class OwlV2(RoboflowInferenceModel): task_type = "object-detection" box_format = "xywh" - def __init__(self, *args, model_id=f"owlv2/{OWLV2_VERSION_ID}", **kwargs): - super().__init__(*args, model_id=model_id, **kwargs) + def __init__(self, model_id=f"owlv2/{OWLV2_VERSION_ID}", *args, **kwargs): + super().__init__(model_id, *args, **kwargs) hf_id = os.path.join("google", self.version_id) processor = Owlv2Processor.from_pretrained(hf_id) self.image_size = tuple(processor.image_processor.size.values())