huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-19 22:02:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Run pre-commit run --all-files to fix CI (#2933)

Browse Source
This commit is contained in:
Alvaro Bartolome 2025-01-21 17:33:33 +01:00 committed by GitHub
parent bdb3e488e4
commit 64a33c1f05
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
5 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
backends/trtllm/scripts/setup_sccache.py
View File

@ -8,9 +8,7 @@ AWS_S3_CACHING_VARIABLES = {
"SCCACHE_BUCKET": "s3_bucket_name", "SCCACHE_BUCKET": "s3_bucket_name",
} }
ALL_CACHING_STORAGE_VARIABLES = { ALL_CACHING_STORAGE_VARIABLES = {"AWS_S3_CACHING_VARIABLES"}
"AWS_S3_CACHING_VARIABLES"
}
def setup_sccache_locally(): def setup_sccache_locally():
@ -29,15 +27,19 @@ def setup_sccache_for_s3():
print("Setting up AWS S3 Caching Layer") print("Setting up AWS S3 Caching Layer")
for envvar in AWS_S3_CACHING_VARIABLES.keys(): for envvar in AWS_S3_CACHING_VARIABLES.keys():
if not envvar in environ or not environ[envvar] or len(environ[envvar]) == 0: if envvar not in environ or not environ[envvar] or len(environ[envvar]) == 0:
print(f"Missing definition for environment variable {envvar}") print(f"Missing definition for environment variable {envvar}")
if __name__ == "__main__": if __name__ == "__main__":
parser = ArgumentParser("TensorRT-LLM Build Caching Setup") parser = ArgumentParser("TensorRT-LLM Build Caching Setup")
parser.add_argument("--is-gha-build", type=str, default="FALSE", parser.add_argument(
help="Indicate if the build is from Github Actions") "--is-gha-build",
type=str,
default="FALSE",
help="Indicate if the build is from Github Actions",
)
# Parse args # Parse args
args = parser.parse_args() args = parser.parse_args()