[FEATS][Tokenizers] [TimmModel] [Odin] [UltralyticsModel]

pull/362/head
Kye 12 months ago
parent e29ec9c943
commit 77ecece6e7

@ -3,7 +3,6 @@ from ultralytics import YOLO
class UltralyticsModel(BaseMultiModalModel): class UltralyticsModel(BaseMultiModalModel):
def __init__(self, model_name: str, *args, **kwargs):
""" """
Initializes an instance of the Ultralytics model. Initializes an instance of the Ultralytics model.
@ -12,6 +11,7 @@ class UltralyticsModel(BaseMultiModalModel):
*args: Variable length argument list. *args: Variable length argument list.
**kwargs: Arbitrary keyword arguments. **kwargs: Arbitrary keyword arguments.
""" """
def __init__(self, model_name: str, *args, **kwargs):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
self.model_name = model_name self.model_name = model_name

Loading…
Cancel
Save