From a06dc6d00facc1d8185897d81a147e23fedf4a87 Mon Sep 17 00:00:00 2001 From: cheetahadmin Date: Thu, 13 Nov 2025 05:10:57 +0000 Subject: [PATCH] Update 1/model.py --- 1/model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/1/model.py b/1/model.py index 6a0787d..ff18379 100644 --- a/1/model.py +++ b/1/model.py @@ -32,7 +32,7 @@ class TritonPythonModel: pretrained_model_name_or_path=self.base_model_path, local_files_only=True, device_map="auto", - trust_remote_code=False + trust_remote_code=True ) # 모델을 평가 모드로 설정 @@ -40,7 +40,7 @@ class TritonPythonModel: self.tokenizer = AutoTokenizer.from_pretrained( self.base_model_path, - trust_remote_code=False, + trust_remote_code=True, add_eos_token=True )