Skip to content

Commit

Permalink
Move import to the top
Browse files Browse the repository at this point in the history
  • Loading branch information
krishung5 committed Oct 6, 2023
1 parent 97abb6b commit 4da060a
Showing 1 changed file with 2 additions and 8 deletions.
10 changes: 2 additions & 8 deletions build.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,8 @@
import stat
import subprocess
import sys
import importlib.util
import requests
from inspect import getsourcefile

#
Expand Down Expand Up @@ -1274,10 +1276,6 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach

if "fastertransformer" in backends:
be = "fastertransformer"
import importlib.util

import requests

url = "https://raw.githubusercontent.com/triton-inference-server/fastertransformer_backend/{}/docker/create_dockerfile_and_build.py".format(
backends[be]
)
Expand Down Expand Up @@ -1318,10 +1316,6 @@ def dockerfile_prepare_container_linux(argmap, backends, enable_gpu, target_mach
# Add dependencies needed for tensorrtllm backend
if "tensorrtllm" in backends:
be = "tensorrtllm"
import importlib.util

import requests

# FIXME: Update the url
url = "https://gitlab-master.nvidia.com/krish/tensorrtllm_backend/-/raw/{}/tools/gen_trtllm_dockerfile.py".format(
backends[be]
Expand Down

0 comments on commit 4da060a

Please sign in to comment.