From 08227101853cbd173104cb1710c1ec0456830bba Mon Sep 17 00:00:00 2001 From: Ariel Kukulanski Date: Thu, 18 Jul 2024 13:22:33 -0300 Subject: [PATCH] Fix NCNN multiple-volumes PNNX download bug (#14533) --- ultralytics/engine/exporter.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ultralytics/engine/exporter.py b/ultralytics/engine/exporter.py index 47466e30..59eb1ae8 100644 --- a/ultralytics/engine/exporter.py +++ b/ultralytics/engine/exporter.py @@ -563,7 +563,7 @@ class Exporter: LOGGER.warning(f"{prefix} WARNING ⚠️ PNNX GitHub assets not found: {e}, using default {asset}") unzip_dir = safe_download(f"https://github.com/pnnx/pnnx/releases/download/{release}/{asset}", delete=True) if check_is_path_safe(Path.cwd(), unzip_dir): # avoid path traversal security vulnerability - (unzip_dir / name).rename(pnnx) # move binary to ROOT + shutil.move(src=unzip_dir / name, dst=pnnx) # move binary to ROOT pnnx.chmod(0o777) # set read, write, and execute permissions for everyone shutil.rmtree(unzip_dir) # delete unzip dir