Skip to content

Commit 6d9ff26

Browse files
authored
fix: 🐛 correct error message for invalid model path in RTDetrDetectionModel (#1266)
Signed-off-by: Onuralp SEZER <thunderbirdtr@gmail.com>
1 parent 7d30708 commit 6d9ff26

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

sahi/models/rtdetr.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
from sahi.models.ultralytics import UltralyticsDetectionModel
44

5+
56
class RTDetrDetectionModel(UltralyticsDetectionModel):
67
def load_model(self):
78
"""Detection model is initialized and set to self.model."""
@@ -13,5 +14,4 @@ def load_model(self):
1314
model.to(self.device)
1415
self.set_model(model)
1516
except Exception as e:
16-
raise TypeError("model_path is not a valid rtdet model path: ", e)
17-
17+
raise TypeError("model_path is not a valid rtdetr model path: ", e)

0 commit comments

Comments
 (0)