From 2060bb58bf5cb11a217a17420ab9a14bc33f818c Mon Sep 17 00:00:00 2001
From: regisss <15324346+regisss@users.noreply.github.com>
Date: Mon, 19 Feb 2024 07:53:24 +0100
Subject: [PATCH] Fix trust remote code (#55)

---
 server/text_generation_server/cli.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/server/text_generation_server/cli.py b/server/text_generation_server/cli.py
index 8e7a63826..90a75f83d 100644
--- a/server/text_generation_server/cli.py
+++ b/server/text_generation_server/cli.py
@@ -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()