From eac3c03f670586e3cd351098dba076a639d81473 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Wed, 1 Dec 2021 20:17:23 +0100
Subject: [PATCH] fix device

---
 hubconf.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hubconf.py b/hubconf.py
index b971abcb2c72..c8cd0a575ba1 100644
--- a/hubconf.py
+++ b/hubconf.py
@@ -43,7 +43,7 @@ def _create(name, pretrained=True, channels=3, classes=80, autoshape=True, verbo
         device = select_device(('0' if torch.cuda.is_available() else 'cpu') if device is None else device)
 
         if pretrained and channels == 3 and classes == 80:
-            model = DetectMultiBackend(path, device='cpu')  # download/load FP32 model
+            model = DetectMultiBackend(path, device=device)  # download/load FP32 model
         else:
             cfg = list((Path(__file__).parent / 'models').rglob(f'{path.name}.yaml'))[0]  # model.yaml path
             model = Model(cfg, channels, classes)  # create model