Improve PNNX download robustness (#9884)

pull/9905/head
Glenn Jocher 7 months ago committed by GitHub
parent c8388d952f
commit bc636d08b7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 8
      ultralytics/engine/exporter.py

@ -529,12 +529,12 @@ class Exporter:
f"or in {ROOT}. See PNNX repo for full installation instructions."
)
system = "macos" if MACOS else "windows" if WINDOWS else "linux-aarch64" if ARM64 else "linux"
_, assets = get_github_assets(repo="pnnx/pnnx", retry=True)
if assets:
try:
_, assets = get_github_assets(repo="pnnx/pnnx", retry=True)
url = [x for x in assets if f"{system}.zip" in x][0]
else:
except Exception as e:
url = f"https://github.com/pnnx/pnnx/releases/download/20240226/pnnx-20240226-{system}.zip"
LOGGER.warning(f"{prefix} WARNING ⚠ PNNX GitHub assets not found, using default {url}")
LOGGER.warning(f"{prefix} WARNING ⚠ PNNX GitHub assets not found: {e}, using default {url}")
asset = attempt_download_asset(url, repo="pnnx/pnnx", release="latest")
if check_is_path_safe(Path.cwd(), asset): # avoid path traversal security vulnerability
unzip_dir = Path(asset).with_suffix("")

Loading…
Cancel
Save