From 46c2a475cb7cf2879b5bed6db96f48cd1930e167 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 2266cab4c0e2..8a4d406f22d9 100644 --- a/hubconf.py +++ b/hubconf.py @@ -14,7 +14,7 @@ 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()