Skip to content

Commit 779cc29

Browse files
Merge pull request #937 from roboflow/fix/OwlV2-init
Fix OwlV2.__init__
2 parents 95a7604 + 31497b0 commit 779cc29

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

inference/models/owlv2/owlv2.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -293,8 +293,8 @@ class OwlV2(RoboflowInferenceModel):
293293
task_type = "object-detection"
294294
box_format = "xywh"
295295

296-
def __init__(self, *args, model_id=f"owlv2/{OWLV2_VERSION_ID}", **kwargs):
297-
super().__init__(*args, model_id=model_id, **kwargs)
296+
def __init__(self, model_id=f"owlv2/{OWLV2_VERSION_ID}", *args, **kwargs):
297+
super().__init__(model_id, *args, **kwargs)
298298
hf_id = os.path.join("google", self.version_id)
299299
processor = Owlv2Processor.from_pretrained(hf_id)
300300
self.image_size = tuple(processor.image_processor.size.values())

0 commit comments

Comments
 (0)