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

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

@ -3,15 +3,15 @@ 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.
Args: Args:
model_name (str): The name of the model. model_name (str): The name of the model.
*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