groupuser/codesage-large-v2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update 1/model.py

Browse Source
This commit is contained in:
cheetahadmin 2025-11-17 08:05:44 +00:00
parent a06dc6d00f
commit 4efc37f7ce
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
1/model.py
View File

@ -31,7 +31,6 @@ class TritonPythonModel:
self.model = AutoModel.from_pretrained( self.model = AutoModel.from_pretrained(
pretrained_model_name_or_path=self.base_model_path, pretrained_model_name_or_path=self.base_model_path,
local_files_only=True, local_files_only=True,
device_map="auto",
trust_remote_code=True trust_remote_code=True
) )