mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-21 23:12:07 +00:00
Fix trust remote code (#55)
This commit is contained in:
parent
2a7a967de3
commit
2060bb58bf
@ -128,6 +128,7 @@ def download_weights(
|
||||
auto_convert: bool = True,
|
||||
logger_level: str = "INFO",
|
||||
json_output: bool = False,
|
||||
trust_remote_code: bool = False,
|
||||
):
|
||||
# Remove default handler
|
||||
logger.remove()
|
||||
|
Loading…
Reference in New Issue
Block a user