tqcq/llama.cpp
0
0
Fork 0
mirror of https://github.com/ggml-org/llama.cpp.git synced 2025-06-27 12:05:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

convert : explicitly disable trust_remote_code for AutoConfig (#13246)

Browse Source
This commit is contained in:
Xuan-Son Nguyen
2025-05-02 08:45:10 +02:00
committed by GitHub
parent d24d592808
commit dcf886007d
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
convert_hf_to_gguf.py
View File

@ -419,7 +419,9 @@ class ModelBase:
@staticmethod @staticmethod
def load_hparams(dir_model: Path): def load_hparams(dir_model: Path):
try: try:
return AutoConfig.from_pretrained(dir_model).to_dict() # for security reason, we don't allow loading remote code by default
# if a model need remote code, we will fallback to config.json
return AutoConfig.from_pretrained(dir_model, trust_remote_code=False).to_dict()
except Exception as e: except Exception as e:
logger.warning(f"Failed to load model config from {dir_model}: {e}") logger.warning(f"Failed to load model config from {dir_model}: {e}")
logger.warning("Trying to load config.json instead") logger.warning("Trying to load config.json instead")