Update model/utils.py

model/utils.py

@@ -9,7 +9,7 @@ from huggingface_hub import hf_hub_download
 import mmengine
 from mmdet3d.apis import init_model, inference_mono_3d_detector
 
-DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
+DEVICE = 'cuda:0' if torch.cuda.is_available() else 'cpu'
 
 MODEL_REGISTRY: Dict[str, Tuple[str, Tuple[str, str]]] = {
     "regnetx4.0gf+detr3d": (