From 4efc37f7ce55ace2f4ee5d929c83d1ddb62839a9 Mon Sep 17 00:00:00 2001 From: cheetahadmin Date: Mon, 17 Nov 2025 08:05:44 +0000 Subject: [PATCH] Update 1/model.py --- 1/model.py | 1 - 1 file changed, 1 deletion(-) diff --git a/1/model.py b/1/model.py index ff18379..b30ffe4 100644 --- a/1/model.py +++ b/1/model.py @@ -31,7 +31,6 @@ class TritonPythonModel: self.model = AutoModel.from_pretrained( pretrained_model_name_or_path=self.base_model_path, local_files_only=True, - device_map="auto", trust_remote_code=True )