Bladeren bron

convert : explicitly disable trust_remote_code for AutoConfig (#13246)

Xuan-Son Nguyen 8 maanden geleden
bovenliggende
commit
dcf886007d
1 gewijzigde bestanden met toevoegingen van 3 en 1 verwijderingen
  1. 3 1
      convert_hf_to_gguf.py

+ 3 - 1
convert_hf_to_gguf.py

@@ -419,7 +419,9 @@ class ModelBase:
     @staticmethod
     def load_hparams(dir_model: Path):
         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:
             logger.warning(f"Failed to load model config from {dir_model}: {e}")
             logger.warning("Trying to load config.json instead")