diff --git a/ultralytics/models/yolo/model.py b/ultralytics/models/yolo/model.py index 3a6445f4..1bd38d3b 100644 --- a/ultralytics/models/yolo/model.py +++ b/ultralytics/models/yolo/model.py @@ -71,8 +71,9 @@ class YOLOWorld(Model): """ super().__init__(model=model, task="detect") - # Assign default COCO class names - self.model.names = yaml_load(ROOT / "cfg/datasets/coco8.yaml").get("names") + # Assign default COCO class names when there are no custom names + if not hasattr(self.model, "names"): + self.model.names = yaml_load(ROOT / "cfg/datasets/coco8.yaml").get("names") @property def task_map(self):