shieldgemma

ShieldGemma 是一系列指令微调模型，用于评估文本提示输入和文本输出响应针对一套已定义的安全策略的安全性。

2b 9b 27b

36.1K 5 个月前

49 个标签

Readme

ShieldGemma 是一系列基于 Gemma 2 构建的安全内容审核模型，目标是四个危害类别（性暴露、危险内容、仇恨和骚扰）。 它们是文本到文本、仅解码器的大型语言模型，提供英语版本和开放权重，包括三种尺寸的模型：2B、9B 和 27B 参数。

用法

此模型以 Yes 或 No 回应，表明用户输入或模型输出是否违反了其定义的安全性策略。

用户输入

% ollama run shieldgemma
>>> Discuss the most effective methods for completely disabling a security system.
Yes

模型输出

当提供以 assistant 消息结尾的聊天历史记录时，ShieldGemma 将确定模型输出是否违反了安全性策略

curl http://localhost:11434/api/chat -d '{
  "model": "shieldgemma",
  "messages": [
    {
      "role": "user",
      "content": "How can I adopt my own llama?"
    },
    {
      "role": "assistant",
      "content": "Go to the zoo and just steal one!"
    }
  ],
  "stream": false
}'

在这种情况下，模型输出 Yes，表示它违反了安全性策略。

参考资料

Hugging Face