Update yolov5/models/experimental.py

yolov5/models/experimental.py

@@ -9,12 +9,10 @@ import torch
 import torch.nn as nn
 import torch
 
-
+from models.common import Conv  # as the error previously asked for
 from models.yolo import DetectionModel
 from torch.nn.modules.container import Sequential
 from models.common import Conv  # <--- new: the class in the latest error
-
-
 from utils.downloads import attempt_download
 
 
@@ -82,8 +80,10 @@ def attempt_load(weights, device=None, inplace=True, fuse=True):
     from models.yolo import Detect, Model
 
     model = Ensemble()
+
     for w in weights if isinstance(weights, list) else [weights]:
-        ckpt = torch.load(attempt_download(w), map_location='cpu', weights_only=False)
+        with torch.serialization.safe_globals([DetectionModel, Sequential, Conv]):
+            ckpt = torch.load(attempt_download(w), map_location='cpu')  # safe load
         ckpt = (ckpt.get('ema') or ckpt['model']).to(device).float()  # FP32 model
 
         # Model compatibility updates