Skip to content

Commit

Permalink
Update default model path for YOLO
Browse files Browse the repository at this point in the history
The old one didn't work.
  • Loading branch information
viklofg committed Apr 10, 2024
1 parent 03c9087 commit 42980ac
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/htrflow_core/models/ultralytics/yolo.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
class YOLO(BaseModel, PytorchDeviceMixin):
def __init__(
self,
model: str | Path = "yolov8n.pt",
model: str | Path = "ultralyticsplus/yolov8s",
device: Optional[str] = None,
cache_dir: str = "./.cache",
hf_token: Optional[str] = None,
Expand Down

0 comments on commit 42980ac

Please sign in to comment.