File tree 1 file changed +3
-2
lines changed
1 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