Update 1/model.py

2025-11-17 08:05:44 +00:00 · 2025-11-17 08:05:44 +00:00 · 4efc37f7ce
commit 4efc37f7ce
parent a06dc6d00f
1 changed files with 0 additions and 1 deletions
--- 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
        )