Procházet zdrojové kódy

convert : explicitly disable trust_remote_code for AutoConfig (#13246)

Xuan-Son Nguyen před 8 měsíci
rodič
revize
dcf886007d
1 změnil soubory, kde provedl 3 přidání a 1 odebrání
  1. 3 1
      convert_hf_to_gguf.py

+ 3 - 1
convert_hf_to_gguf.py

@@ -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")