Commit
·
7816c0c
1
Parent(s):
93fc607
PyTorch Hub and autoShape update (#1415)
Browse files- hubconf.py +1 -1
hubconf.py
CHANGED
@@ -14,7 +14,7 @@ from models.yolo import Model
|
|
14 |
from utils.general import set_logging
|
15 |
from utils.google_utils import attempt_download
|
16 |
|
17 |
-
dependencies = ['torch', '
|
18 |
set_logging()
|
19 |
|
20 |
|
|
|
14 |
from utils.general import set_logging
|
15 |
from utils.google_utils import attempt_download
|
16 |
|
17 |
+
dependencies = ['torch', 'yaml', 'Pillow']
|
18 |
set_logging()
|
19 |
|
20 |
|