File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change 77import shutil
88import subprocess
99import sys
10+ import urllib .request
1011import warnings
1112from dataclasses import fields , replace
1213from enum import Enum
@@ -827,10 +828,10 @@ def download_plugin_lib_path(py_version: str, platform: str) -> str:
827828 base_url = "https://pypi.nvidia.com/tensorrt-llm/"
828829 file_name = f"tensorrt_llm-0.18.0-{ py_version } -{ py_version } -{ platform } .whl"
829830 download_url = base_url + file_name
830- cmd = ["wget" , download_url ]
831831 if not (os .path .exists (file_name )):
832832 try :
833- subprocess .run (cmd , check = True )
833+ logger .debug (f"Downloading { download_url } ..." )
834+ urllib .request .urlretrieve (download_url , file_name )
834835 logger .debug ("Download succeeded and TRT-LLM wheel is now present" )
835836 except subprocess .CalledProcessError as e :
836837 logger .error (
You can’t perform that action at this time.
0 commit comments