We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
2 parents 95a7604 + 31497b0 commit 779cc29Copy full SHA for 779cc29
inference/models/owlv2/owlv2.py
@@ -293,8 +293,8 @@ class OwlV2(RoboflowInferenceModel):
293
task_type = "object-detection"
294
box_format = "xywh"
295
296
- def __init__(self, *args, model_id=f"owlv2/{OWLV2_VERSION_ID}", **kwargs):
297
- 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)
298
hf_id = os.path.join("google", self.version_id)
299
processor = Owlv2Processor.from_pretrained(hf_id)
300
self.image_size = tuple(processor.image_processor.size.values())
0 commit comments