trust_remote_code

This commit is contained in:
游雁 2025-08-15 15:10:37 +08:00
parent 5115a066c9
commit f5051c55cd

View File

@ -1,3 +1,4 @@
import logging
import os
import json
from omegaconf import OmegaConf, DictConfig
@ -79,7 +80,10 @@ def download_from_ms(**kwargs):
kwargs["jieba_usr_dict"] = os.path.join(model_or_path, "jieba_usr_dict")
if isinstance(kwargs, DictConfig):
kwargs = OmegaConf.to_container(kwargs, resolve=True)
if os.path.exists(os.path.join(model_or_path, "requirements.txt")):
logging.warning(f'trust_remote_code: {kwargs.get("trust_remote_code", False)}')
if os.path.exists(os.path.join(model_or_path, "requirements.txt")) and kwargs.get(
"trust_remote_code", False
):
requirements = os.path.join(model_or_path, "requirements.txt")
print(f"Detect model requirements, begin to install it: {requirements}")
from funasr.utils.install_model_requirements import install_requirements