diff --git a/hubconf.py b/hubconf.py index 8e06beb..5cc3ded 100644 --- a/hubconf.py +++ b/hubconf.py @@ -5,7 +5,7 @@ Usage: model = torch.hub.load('ultralytics/yolov5', 'yolov5s', pretrained=True, channels=3, classes=80) """ -dependencies = ['torch', 'pyyaml'] +dependencies = ['torch', 'yaml'] import torch from models.yolo import Model