From 7816c0c2d3d12b9919aa8b9a25125714b00d5881 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Mon, 16 Nov 2020 23:15:43 +0100 Subject: [PATCH] PyTorch Hub and autoShape update (#1415) --- hubconf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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()