diff --git a/hubconf.py b/hubconf.py index 2266cab..8a4d406 100644 --- a/hubconf.py +++ b/hubconf.py @@ -14,7 +14,7 @@ from models.yolo import Model from utils.general import set_logging from utils.google_utils import attempt_download -dependencies = ['torch', 'PyYAML>=5.3', 'Pillow'] +dependencies = ['torch', 'yaml', 'Pillow'] set_logging()