From 77ecece6e79857ba2ae54969390b6f2391205052 Mon Sep 17 00:00:00 2001 From: Kye Date: Sat, 20 Jan 2024 00:26:47 -0500 Subject: [PATCH] [FEATS][Tokenizers] [TimmModel] [Odin] [UltralyticsModel] --- swarms/models/ultralytics_model.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/swarms/models/ultralytics_model.py b/swarms/models/ultralytics_model.py index 2df548f4..2ce619d2 100644 --- a/swarms/models/ultralytics_model.py +++ b/swarms/models/ultralytics_model.py @@ -3,15 +3,15 @@ from ultralytics import YOLO 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: - model_name (str): The name of the model. - *args: Variable length argument list. - **kwargs: Arbitrary keyword arguments. - """ + Args: + model_name (str): The name of the model. + *args: Variable length argument list. + **kwargs: Arbitrary keyword arguments. + """ + def __init__(self, model_name: str, *args, **kwargs): super().__init__(*args, **kwargs) self.model_name = model_name