无法连接Hugging Face加载安全检查器模型的问题求助

问题根源 #

脚本执行时尝试从Hugging Face Hub加载CompVis/stable-diffusion-safety-checker的特征提取器，但网络连接失败，本地缓存无对应文件，且未指定有效的本地模型目录，导致报错。

解决方案 #

• 修复网络连接并配置镜像（推荐）

  • 确认本地网络可正常访问Hugging Face相关服务
  • 若处于国内网络环境，在终端执行以下命令配置镜像源：

    export HF_ENDPOINT=https://hf-mirror.com
    

  • 配置完成后重新运行run_features_extraction.py

• 手动下载模型到本地

  • 手动获取CompVis/stable-diffusion-safety-checker的全部文件，重点确保包含preprocessor_config.json
  • 将文件存放到本地指定目录，例如./local-safety-checker/
  • 修改pnp_utils.py中的代码，将模型加载路径替换为本地目录：

    safety_feature_extractor = AutoFeatureExtractor.from_pretrained("./local-safety-checker/")
    

  • 保存修改后重新执行脚本

• 临时禁用安全检查模块（仅测试用）

  • 注释掉run_features_extraction.py中from pnp_utils import check_safety的语句
  • 同时注释掉pnp_utils.py中加载safety_feature_extractor的相关代码
  • 注意：此操作会关闭内容安全校验，请勿在正式生产环境使用

内容的提问来源于stack exchange，提问作者Our bank