From e61d13acdf3193606c3bc57fb59f0de33eab7490 Mon Sep 17 00:00:00 2001 From: Ying Sheng Date: Sat, 24 Aug 2024 18:35:55 -0700 Subject: [PATCH] [CI] Fix the problem of hf runner too slow (#1202) --- python/sglang/test/runners.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/python/sglang/test/runners.py b/python/sglang/test/runners.py index e519c9282..4fc1f0f25 100644 --- a/python/sglang/test/runners.py +++ b/python/sglang/test/runners.py @@ -86,7 +86,6 @@ class HFRunner: self.tokenizer = AutoTokenizer.from_pretrained( model_path, torch_dtype=torch_dtype, - trust_remote_code=True, ) self.is_generation_model = is_generation_model @@ -96,7 +95,6 @@ class HFRunner: model_path, torch_dtype=torch_dtype, low_cpu_mem_usage=True, - trust_remote_code=True, ).cuda() else: from sentence_transformers import SentenceTransformer