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

Fix trust remote code (#55)

Browse Source
This commit is contained in:
regisss 2024-02-19 07:53:24 +01:00 committed by GitHub
parent 2a7a967de3
commit 2060bb58bf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
server/text_generation_server/cli.py
View File

@ -128,6 +128,7 @@ def download_weights(
auto_convert: bool = True, auto_convert: bool = True,
logger_level: str = "INFO", logger_level: str = "INFO",
json_output: bool = False, json_output: bool = False,
trust_remote_code: bool = False,
): ):
# Remove default handler # Remove default handler
logger.remove() logger.remove()