File tree
Expand file treeCollapse file tree1 file changed
+3
-2
lines changed Expand file treeCollapse file tree1 file changed
+3
-2
lines changed Original file line number | Diff line number | Diff line change |
---|
|
7 | 7 | import shutil
|
8 | 8 | import subprocess
|
9 | 9 | import sys
|
| 10 | +import urllib.request |
10 | 11 | import warnings
|
11 | 12 | from dataclasses import fields, replace
|
12 | 13 | from enum import Enum
|
@@ -827,10 +828,10 @@ def download_plugin_lib_path(py_version: str, platform: str) -> str:
|
827 | 828 | base_url = "https://pypi.nvidia.com/tensorrt-llm/"
|
828 | 829 | file_name = f"tensorrt_llm-0.18.0-{py_version}-{py_version}-{platform}.whl"
|
829 | 830 | download_url = base_url + file_name
|
830 |
| -cmd = ["wget", download_url] |
831 | 831 | if not (os.path.exists(file_name)):
|
832 | 832 | try:
|
833 |
| -subprocess.run(cmd, check=True) |
| 833 | +logger.debug(f"Downloading {download_url} ...") |
| 834 | +urllib.request.urlretrieve(download_url, file_name) |
834 | 835 | logger.debug("Download succeeded and TRT-LLM wheel is now present")
|
835 | 836 | except subprocess.CalledProcessError as e:
|
836 | 837 | logger.error(
|
|
You can’t perform that action at this time.
0 commit comments