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

2b 9b 27b

8,164 4 周前

自述文件

ShieldGemma 是一系列基于 Gemma 2 的安全内容审核模型,针对四个伤害类别(色情内容、危险内容、仇恨和骚扰)。它们是文本到文本、仅解码器的大型语言模型,提供英语版本并公开权重,包括三种大小的模型:2B、9B 和 27B 个参数。

用法

该模型会以 回答用户输入或模型输出是否违反其定义的安全策略。

用户输入

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

模型输出

当提供一个以 助理 消息结尾的聊天记录时,ShieldGemma 将确定模型输出是否违反安全策略。

curl https://127.0.0.1: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
}'

在这种情况下,模型输出 ,意味着它违反了安全策略。

参考资料

Hugging Face