diff --git a/yolort/models/yolo.py b/yolort/models/yolo.py index 2d37b3c9..80471377 100644 --- a/yolort/models/yolo.py +++ b/yolort/models/yolo.py @@ -273,8 +273,12 @@ def build_model( return model -model_urls_root_r30 = "https://github.com/zhiqwang/yolov5-rt-stack/releases/download/v0.3.0" -model_urls_root_r52 = "https://github.com/zhiqwang/yolov5-rt-stack/releases/download/v0.5.2-alpha" +model_urls_root_r30 = ( + "https://github.com/zhiqwang/yolov5-rt-stack/releases/download/v0.3.0" +) +model_urls_root_r52 = ( + "https://github.com/zhiqwang/yolov5-rt-stack/releases/download/v0.5.2-alpha" +) model_urls = { # Path Aggregation Network 3.1 and 4.0