Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,20 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+Q2_K/Q2_K-00001-of-00001.gguf filter=lfs diff=lfs merge=lfs -text
+Q3_K/Q3_K-00001-of-00001.gguf filter=lfs diff=lfs merge=lfs -text
+Q3_K_L/Q3_K_L-00001-of-00001.gguf filter=lfs diff=lfs merge=lfs -text
+Q3_K_M/Q3_K_M-00001-of-00001.gguf filter=lfs diff=lfs merge=lfs -text
+Q3_K_S/Q3_K_S-00001-of-00001.gguf filter=lfs diff=lfs merge=lfs -text
+Q4_0/Q4_0-00001-of-00001.gguf filter=lfs diff=lfs merge=lfs -text
+Q4_1/Q4_1-00001-of-00001.gguf filter=lfs diff=lfs merge=lfs -text
+Q4_K/Q4_K-00001-of-00001.gguf filter=lfs diff=lfs merge=lfs -text
+Q4_K_M/Q4_K_M-00001-of-00001.gguf filter=lfs diff=lfs merge=lfs -text
+Q4_K_S/Q4_K_S-00001-of-00001.gguf filter=lfs diff=lfs merge=lfs -text
+Q5_0/Q5_0-00001-of-00001.gguf filter=lfs diff=lfs merge=lfs -text
+Q5_1/Q5_1-00001-of-00001.gguf filter=lfs diff=lfs merge=lfs -text
+Q5_K/Q5_K-00001-of-00001.gguf filter=lfs diff=lfs merge=lfs -text
+Q5_K_M/Q5_K_M-00001-of-00001.gguf filter=lfs diff=lfs merge=lfs -text
+Q5_K_S/Q5_K_S-00001-of-00001.gguf filter=lfs diff=lfs merge=lfs -text
+Q6_K/Q6_K-00001-of-00001.gguf filter=lfs diff=lfs merge=lfs -text
+Q8_0/Q8_0-00001-of-00001.gguf filter=lfs diff=lfs merge=lfs -text

Q2_K/Q2_K-00001-of-00001.gguf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:27a77fcc40a3bfdfaad29f5ba40da15ba7a704cd33c1ac7ffb16aa107eaf812d
+size 1229830400

Q3_K/Q3_K-00001-of-00001.gguf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0a224f1c69c3997c52e73be8bb25ab3caef8b99137fcfc30fcbba12f74e8ae34
+size 1461668096

Q3_K_L/Q3_K_L-00001-of-00001.gguf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:319fc057d0111bc62b1964cbddacf93e99b814f220899772cd74845eceec7882
+size 1550436608

Q3_K_M/Q3_K_M-00001-of-00001.gguf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0a224f1c69c3997c52e73be8bb25ab3caef8b99137fcfc30fcbba12f74e8ae34
+size 1461668096

Q3_K_S/Q3_K_S-00001-of-00001.gguf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:efe5f41638d1a2ad2b0ec3e30dd2bba4e27f920c43b4a7968516cf1d98ddc7d6
+size 1360660736

Q4_0/Q4_0-00001-of-00001.gguf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4d08bd6dfa1fb31b01ac8ccbb2021d63925ca4ce0f8cbf6c5d407b2fd5e21b7e
+size 1629509888

Q4_1/Q4_1-00001-of-00001.gguf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:da7151ee9aba9c336ae5e4c9616aa143f6755f9f0348706a974b7a84973d57bd
+size 1756027136

Q4_K/Q4_K-00001-of-00001.gguf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:850bd680372bf94cb36772b110e9082ae4fb8623c468bd6ec57bcf58f92d7eaa
+size 1708583168

Q4_K_M/Q4_K_M-00001-of-00001.gguf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:850bd680372bf94cb36772b110e9082ae4fb8623c468bd6ec57bcf58f92d7eaa
+size 1708583168

Q4_K_S/Q4_K_S-00001-of-00001.gguf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6ffb2df0351c4b75444eb5ffbdcdb14a209bf24ad3a369c30601dbe2bc319b19
+size 1638652160

Q5_0/Q5_0-00001-of-00001.gguf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1cce4f5d81a49d0a80934b686f02460e50fdecbbee92dd1c9cf9ea7ff0316f07
+size 1882544384

Q5_1/Q5_1-00001-of-00001.gguf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bf30d516cf9a10e79df68f14bf6bd869a4fa925c0d894d4f2c2f18b22badae7f
+size 2009061632

Q5_K/Q5_K-00001-of-00001.gguf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:168240a8ed46fff6b2d493a691f6d1099c4f83c79af68a473aa14f5b32a79239
+size 1923279104

Q5_K_M/Q5_K_M-00001-of-00001.gguf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:168240a8ed46fff6b2d493a691f6d1099c4f83c79af68a473aa14f5b32a79239
+size 1923279104

Q5_K_S/Q5_K_S-00001-of-00001.gguf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4dfe8cf04a47b46f9938db93e7ef49b0e9644aed93f6083c1ac3e1169b20634a
+size 1882544384

Q6_K/Q6_K-00001-of-00001.gguf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f35a73978acba3b093687a5fbe6cfdaca9f1a6a1bb618463787fa5e8cfe01ee2
+size 2151393536

Q8_0/Q8_0-00001-of-00001.gguf

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d8d2f6ba53ea4eb24fc0b740c25fc607fb49f76cf86c7ab8d9fa22a6c37213e8
+size 2784495872

README.md

@@ -0,0 +1,526 @@
+---
+library_name: transformers
+license: gemma
+pipeline_tag: text-generation
+tags:
+- GGUF
+extra_gated_heading: Access Gemma on Hugging Face
+extra_gated_prompt: To access Gemma on Hugging Face, you’re required to review and
+  agree to Google’s usage license. To do this, please ensure you’re logged in to Hugging
+  Face and click below. Requests are processed immediately.
+extra_gated_button_content: Acknowledge license
+quantized_by: andrijdavid
+---
+# shieldgemma-2b-GGUF
+- Original model: [shieldgemma-2b](https://huggingface.co/google/shieldgemma-2b)
+
+<!-- description start -->
+## Description
+
+This repo contains GGUF format model files for [shieldgemma-2b](https://huggingface.co/google/shieldgemma-2b).
+
+<!-- description end -->
+<!-- README_GGUF.md-about-gguf start -->
+### About GGUF
+GGUF is a new format introduced by the llama.cpp team on August 21st 2023. It is a replacement for GGML, which is no longer supported by llama.cpp.
+Here is an incomplete list of clients and libraries that are known to support GGUF:
+* [llama.cpp](https://github.com/ggerganov/llama.cpp). This is the source project for GGUF, providing both a Command Line Interface (CLI) and a server option.
+* [text-generation-webui](https://github.com/oobabooga/text-generation-webui), Known as the most widely used web UI, this project boasts numerous features and powerful extensions, and supports GPU acceleration.
+* [Ollama](https://github.com/jmorganca/ollama) Ollama is a lightweight and extensible framework designed for building and running language models locally. It features a simple API for creating, managing, and executing models, along with a library of pre-built models for use in various applications​
+* [KoboldCpp](https://github.com/LostRuins/koboldcpp), A comprehensive web UI offering GPU acceleration across all platforms and architectures, particularly renowned for storytelling.
+* [GPT4All](https://gpt4all.io), This is a free and open source GUI that runs locally, supporting Windows, Linux, and macOS with full GPU acceleration.
+* [LM Studio](https://lmstudio.ai/) An intuitive and powerful local GUI for Windows and macOS (Silicon), featuring GPU acceleration.
+* [LoLLMS Web UI](https://github.com/ParisNeo/lollms-webui). A notable web UI with a variety of unique features, including a comprehensive model library for easy model selection.
+* [Faraday.dev](https://faraday.dev/), An attractive, user-friendly character-based chat GUI for Windows and macOS (both Silicon and Intel), also offering GPU acceleration.
+* [llama-cpp-python](https://github.com/abetlen/llama-cpp-python),  A Python library equipped with GPU acceleration, LangChain support, and an OpenAI-compatible API server.
+* [candle](https://github.com/huggingface/candle), A Rust-based ML framework focusing on performance, including GPU support, and designed for ease of use.
+* [ctransformers](https://github.com/marella/ctransformers), A Python library featuring GPU acceleration, LangChain support, and an OpenAI-compatible AI server.
+* [localGPT](https://github.com/PromtEngineer/localGPT) An open-source initiative enabling private conversations with documents. 
+<!-- README_GGUF.md-about-gguf end -->
+
+<!-- compatibility_gguf start -->
+## Explanation of quantisation methods
+<details>
+  <summary>Click to see details</summary>
+The new methods available are:
+
+* GGML_TYPE_Q2_K - "type-1" 2-bit quantization in super-blocks containing 16 blocks, each block having 16 weight. Block scales and mins are quantized with 4 bits. This ends up effectively using 2.5625 bits per weight (bpw)
+* GGML_TYPE_Q3_K - "type-0" 3-bit quantization in super-blocks containing 16 blocks, each block having 16 weights. Scales are quantized with 6 bits. This end up using 3.4375 bpw.
+* GGML_TYPE_Q4_K - "type-1" 4-bit quantization in super-blocks containing 8 blocks, each block having 32 weights. Scales and mins are quantized with 6 bits. This ends up using 4.5 bpw.
+* GGML_TYPE_Q5_K - "type-1" 5-bit quantization. Same super-block structure as GGML_TYPE_Q4_K resulting in 5.5 bpw
+* GGML_TYPE_Q6_K - "type-0" 6-bit quantization. Super-blocks with 16 blocks, each block having 16 weights. Scales are quantized with 8 bits. This ends up using 6.5625 bpw.
+</details>
+<!-- compatibility_gguf end -->
+
+<!-- README_GGUF.md-how-to-download start -->
+## How to download GGUF files
+
+**Note for manual downloaders:** You almost never want to clone the entire repo! Multiple different quantisation formats are provided, and most users only want to pick and download a single folder.
+
+The following clients/libraries will automatically download models for you, providing a list of available models to choose from:
+
+* LM Studio
+* LoLLMS Web UI
+* Faraday.dev
+
+### In `text-generation-webui`
+
+Under Download Model, you can enter the model repo: LiteLLMs/shieldgemma-2b-GGUF and below it, a specific filename to download, such as: Q4_0/Q4_0-00001-of-00001.gguf.
+
+Then click Download.
+
+### On the command line, including multiple files at once
+
+I recommend using the `huggingface-hub` Python library:
+
+```shell
+pip3 install huggingface-hub
+```
+
+Then you can download any individual model file to the current directory, at high speed, with a command like this:
+
+```shell
+huggingface-cli download LiteLLMs/shieldgemma-2b-GGUF Q4_0/Q4_0-00001-of-00001.gguf --local-dir . --local-dir-use-symlinks False
+```
+
+<details>
+  <summary>More advanced huggingface-cli download usage (click to read)</summary>
+
+You can also download multiple files at once with a pattern:
+
+```shell
+huggingface-cli download LiteLLMs/shieldgemma-2b-GGUF --local-dir . --local-dir-use-symlinks False --include='*Q4_K*gguf'
+```
+
+For more documentation on downloading with `huggingface-cli`, please see: [HF -> Hub Python Library -> Download files -> Download from the CLI](https://huggingface.co/docs/huggingface_hub/guides/download#download-from-the-cli).
+
+To accelerate downloads on fast connections (1Gbit/s or higher), install `hf_transfer`:
+
+```shell
+pip3 install huggingface_hub[hf_transfer]
+```
+
+And set environment variable `HF_HUB_ENABLE_HF_TRANSFER` to `1`:
+
+```shell
+HF_HUB_ENABLE_HF_TRANSFER=1 huggingface-cli download LiteLLMs/shieldgemma-2b-GGUF Q4_0/Q4_0-00001-of-00001.gguf --local-dir . --local-dir-use-symlinks False
+```
+
+Windows Command Line users: You can set the environment variable by running `set HF_HUB_ENABLE_HF_TRANSFER=1` before the download command.
+</details>
+<!-- README_GGUF.md-how-to-download end -->
+<!-- README_GGUF.md-how-to-run start -->
+## Example `llama.cpp` command
+
+Make sure you are using `llama.cpp` from commit [d0cee0d](https://github.com/ggerganov/llama.cpp/commit/d0cee0d36d5be95a0d9088b674dbb27354107221) or later.
+
+```shell
+./main -ngl 35 -m Q4_0/Q4_0-00001-of-00001.gguf --color -c  --temp 0.7 --repeat_penalty 1.1 -n -1 -p "<PROMPT>"
+```
+
+Change `-ngl 32` to the number of layers to offload to GPU. Remove it if you don't have GPU acceleration.
+
+Change `-c ` to the desired sequence length. For extended sequence models - eg 8K, 16K, 32K - the necessary RoPE scaling parameters are read from the GGUF file and set by llama.cpp automatically. Note that longer sequence lengths require much more resources, so you may need to reduce this value.
+
+If you want to have a chat-style conversation, replace the `-p <PROMPT>` argument with `-i -ins`
+
+For other parameters and how to use them, please refer to [the llama.cpp documentation](https://github.com/ggerganov/llama.cpp/blob/master/examples/main/README.md)
+
+## How to run in `text-generation-webui`
+
+Further instructions can be found in the text-generation-webui documentation, here: [text-generation-webui/docs/04 ‐ Model Tab.md](https://github.com/oobabooga/text-generation-webui/blob/main/docs/04%20%E2%80%90%20Model%20Tab.md#llamacpp).
+
+## How to run from Python code
+
+You can use GGUF models from Python using the [llama-cpp-python](https://github.com/abetlen/llama-cpp-python) or [ctransformers](https://github.com/marella/ctransformers) libraries. Note that at the time of writing (Nov 27th 2023), ctransformers has not been updated for some time and is not compatible with some recent models. Therefore I recommend you use llama-cpp-python.
+
+### How to load this model in Python code, using llama-cpp-python
+
+For full documentation, please see: [llama-cpp-python docs](https://abetlen.github.io/llama-cpp-python/).
+
+#### First install the package
+
+Run one of the following commands, according to your system:
+
+```shell
+# Base ctransformers with no GPU acceleration
+pip install llama-cpp-python
+# With NVidia CUDA acceleration
+CMAKE_ARGS="-DLLAMA_CUBLAS=on" pip install llama-cpp-python
+# Or with OpenBLAS acceleration
+CMAKE_ARGS="-DLLAMA_BLAS=ON -DLLAMA_BLAS_VENDOR=OpenBLAS" pip install llama-cpp-python
+# Or with CLBLast acceleration
+CMAKE_ARGS="-DLLAMA_CLBLAST=on" pip install llama-cpp-python
+# Or with AMD ROCm GPU acceleration (Linux only)
+CMAKE_ARGS="-DLLAMA_HIPBLAS=on" pip install llama-cpp-python
+# Or with Metal GPU acceleration for macOS systems only
+CMAKE_ARGS="-DLLAMA_METAL=on" pip install llama-cpp-python
+# In windows, to set the variables CMAKE_ARGS in PowerShell, follow this format; eg for NVidia CUDA:
+$env:CMAKE_ARGS = "-DLLAMA_OPENBLAS=on"
+pip install llama-cpp-python
+```
+
+#### Simple llama-cpp-python example code
+
+```python
+from llama_cpp import Llama
+# Set gpu_layers to the number of layers to offload to GPU. Set to 0 if no GPU acceleration is available on your system.
+llm = Llama(
+  model_path="./Q4_0/Q4_0-00001-of-00001.gguf",  # Download the model file first
+  n_ctx=32768,  # The max sequence length to use - note that longer sequence lengths require much more resources
+  n_threads=8,            # The number of CPU threads to use, tailor to your system and the resulting performance
+  n_gpu_layers=35         # The number of layers to offload to GPU, if you have GPU acceleration available
+)
+# Simple inference example
+output = llm(
+  "<PROMPT>", # Prompt
+  max_tokens=512,  # Generate up to 512 tokens
+  stop=["</s>"],   # Example stop token - not necessarily correct for this specific model! Please check before using.
+  echo=True        # Whether to echo the prompt
+)
+# Chat Completion API
+llm = Llama(model_path="./Q4_0/Q4_0-00001-of-00001.gguf", chat_format="llama-2")  # Set chat_format according to the model you are using
+llm.create_chat_completion(
+    messages = [
+        {"role": "system", "content": "You are a story writing assistant."},
+        {
+            "role": "user",
+            "content": "Write a story about llamas."
+        }
+    ]
+)
+```
+
+## How to use with LangChain
+
+Here are guides on using llama-cpp-python and ctransformers with LangChain:
+
+* [LangChain + llama-cpp-python](https://python.langchain.com/docs/integrations/llms/llamacpp)
+* [LangChain + ctransformers](https://python.langchain.com/docs/integrations/providers/ctransformers)
+
+<!-- README_GGUF.md-how-to-run end -->
+
+<!-- footer end -->
+
+<!-- original-model-card start -->
+# Original model card: shieldgemma-2b
+
+
+# ShieldGemma model card
+
+**Model Page**: [ShieldGemma][shieldgemma]
+
+**Resources and Technical Documentation**:
+
+* [Responsible Generative AI Toolkit][rai-toolkit]
+* [ShieldGemma on Kaggle][shieldgemma-kaggle]
+* [ShieldGemma on Hugging Face Hub][shieldgemma-hfhub]
+
+**Terms of Use**: [Terms][terms]
+
+**Authors**: Google
+
+## Model Information
+
+Summary description and brief definition of inputs and outputs.
+
+### Description
+
+ShieldGemma is a series of safety content moderation models built upon
+[Gemma 2][gemma2] that target four harm categories (sexually explicit, dangerous
+content, hate, and harassment). They are text-to-text, decoder-only large
+language models, available in English with open weights, including models of 3
+sizes: 2B, 9B and 27B parameters.
+
+### Inputs and outputs
+
+*   **Input:** Text string containing a preamble, the text to be classified, a
+    set of policies, and the prompt epilogue. The full prompt must be formatted
+    using a specific pattern for optimal performance. The pattern used for the
+    reported evaluation metrics is described in this section.
+*   **Output:** Text string, which will start with the token "Yes" or "No" and
+    represent whether the user input or model output violates the provided
+    policies.
+
+The prompt pattern contains the following components, in order:
+
+1.  Preamble, establishing the model as a policy expert, based on the
+    [LLM-as-a-judge][llm-as-judge] technique.
+1.  User prompt, wrapped with `<start_of_turn>` and `<end_of_turn>` control
+    tokens.
+1.  Description of the safety policy.
+1.  Optionally, a model response also wrapped by `<start_of_turn>` and
+    `<end_of_turn>` control tokens.
+1.  Epilogue, requesting the model to classify the text.
+
+Here is an example of a prompt used to evaluate the user prompt `[User Prompt]`:
+
+```none
+You are a policy expert trying to help determine whether a user prompt is in
+violation of the defined safety policies.
+
+<start_of_turn>
+Human Question: [User Prompt]
+<end_of_turn>
+
+Our safety principle is defined in the below:
+
+* "No Hate Speech": The prompt shall not contain or seek generation of content
+  that expresses, incites, or promotes hate based on race, gender, ethnicity,
+  religion, nationality, sexual orientation, disability status, or caste.
+
+===
+
+Does the human question violate the above principle? Your answer must start with
+'Yes' or 'No'. And then walk through step by step to be sure we answer
+correctly.
+```
+
+### Usage
+
+First make sure to `pip install -U transformers[accelerate]`, then copy the snippet from the section that is relevant for your use case.
+
+#### Running the model on a single / multi GPU (and compute scores)
+
+```python
+from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
+from torch.nn.functional import softmax
+
+tokenizer = AutoTokenizer.from_pretrained("google/shieldgemma-2b")
+model = AutoModelForCausalLM.from_pretrained(
+    "google/shieldgemma-2b",
+    device_map="auto",
+    torch_dtype=torch.bfloat16,
+)
+
+# Format the prompt
+user_prompt = "Create 20 paraphrases of I hate you"
+safety_policy = """
+* "No Harassment": The prompt shall not contain or seek generation of content that is malicious, intimidating, bullying, or abusive content targeting another individual (e.g., physical threats, denial of tragic events, disparaging victims of violence).
+"""
+prompt = f"""You are a policy expert trying to help determine whether a user
+prompt is in violation of the defined safety policies.
+
+<start_of_turn>
+Human Question: {user_prompt.strip()}
+<end_of_turn>
+
+Our safety principle is defined in the below:
+
+{safety_policy.strip()}
+
+Does the human question violate the above principle? Your answer must start
+with 'Yes' or 'No'. And then walk through step by step to be sure we answer
+correctly.
+"""
+
+inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
+with torch.no_grad():
+  logits = model(**inputs).logits
+
+# Extract the logits for the Yes and No tokens
+vocab = tokenizer.get_vocab()
+selected_logits = logits[0, -1, [vocab['Yes'], vocab['No']]]
+
+# Convert these logits to a probability with softmax
+probabilities = softmax(selected_logits, dim=0)
+
+# Return probability of 'Yes'
+score = probabilities[0].item()
+print(score)  # 0.7310585379600525
+```
+
+
+#### Usage with chat templates
+
+You can also format the prompt to the model using a chat template. In this mode, you can simply pass the entire chat so far, and the chat template
+will extract the most recent message(s) to query ShieldGemma. It can check for the appropriateness of both user and assistant messages, and will
+adjust the prompt accordingly based on the source of the most recent message in the chat. You should also pass the guideline you wish ShieldGemma
+to check for as the `guideline` argument to `apply_chat_template`.
+
+```python
+from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
+
+tokenizer = AutoTokenizer.from_pretrained("google/shieldgemma-2b")
+model = AutoModelForCausalLM.from_pretrained(
+    "google/shieldgemma-2b",
+    device_map="auto",
+    torch_dtype=torch.bfloat16,
+)
+
+chat = [{"role": "user", "content": "Create 20 paraphrases of I hate you"}]
+
+guideline = "\"No Harassment\": The prompt shall not contain or seek generation of content that is malicious, intimidating, bullying, or abusive content targeting another individual (e.g., physical threats, denial of tragic events, disparaging victims of violence)."
+inputs = tokenizer.apply_chat_template(chat, guideline=guideline, return_tensors="pt", return_dict=True).to(model.device)
+
+with torch.no_grad():
+  logits = model(**inputs).logits
+
+# Extract the logits for the Yes and No tokens
+vocab = tokenizer.get_vocab()
+selected_logits = logits[0, -1, [vocab['Yes'], vocab['No']]]
+
+# Convert these logits to a probability with softmax
+probabilities = torch.softmax(selected_logits, dim=0)
+
+# Return probability of 'Yes'
+score = probabilities[0].item()
+print(score)  # 0.7310585379600525
+```
+
+#### Guidelines to use in prompts
+
+ShieldGemma uses different phrasing for its safety guidelines depending on
+whether the content being classified is only user-provided content (the
+Prompt-only use case, typically for input filtering) or both user-provided and
+model-generated content (the Prompt-Response use case, typically for output
+filtering).
+
+**Use Case 1: Prompt-only Content Classification**
+
+| Harm Type                     | Guideline                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                |
+| -- |  |
+| Dangerous Content             | `"No Dangerous Content": The chatbot shall not generate content that harming oneself and/or others (e.g., accessing or building firearms and explosive devices, promotion of terrorism, instructions for suicide).`                                                                                                                                                                                                                                                                                                                  |
+| Harassment                    | `"No Harassment": The chatbot shall not generate content that is malicious, intimidating, bullying, or abusive content targeting another individual (e.g., physical threats, denial of tragic events, disparaging victims of violence).`                                                                                                                                                                                                                                                                                             |
+| Hate Speech                   | `"No Hate Speech": The chatbot shall not generate content that targeting identity and/or protected attributes (e.g., racial slurs, promotion of discrimination, calls to violence against protected groups) or dehumanizing/belittling/vilifying on the bases of race, ethnicity, religion, disability, age, nationality, veteran status, sexual orientation, gender, gender identity, caste, or any other protected status. We do allow content that includes a reference to a protected individual or group in a positive manner.` |
+| Sexually Explicit Information | `"No Sexually Explicit Information": The chatbot shall not generate content that contains references to sexual acts or other lewd content (e.g., sexually graphic descriptions, content aimed at causing arousal). Medical or scientific terms related to human anatomy or sex education are permitted.`                                                                                                                                                                                                                             |
+
+### Citation
+
+```plaintext
+@misc{zeng2024shieldgemmagenerativeaicontent,
+      title={ShieldGemma: Generative AI Content Moderation Based on Gemma}, 
+      author={Wenjun Zeng and Yuchi Liu and Ryan Mullins and Ludovic Peran and Joe Fernandez and Hamza Harkous and Karthik Narasimhan and Drew Proud and Piyush Kumar and Bhaktipriya Radharapu and Olivia Sturman and Oscar Wahltinez},
+      year={2024},
+      eprint={2407.21772},
+      archivePrefix={arXiv},
+      primaryClass={cs.CL},
+      url={https://arxiv.org/abs/2407.21772}, 
+}
+```
+
+## Model Data
+
+Data used for model training and how the data was processed.
+
+### Training Dataset
+
+The base models were trained on a dataset of text data that includes a wide
+variety of sources, see the [Gemma 2 documentation][gemma2] for more details. The
+ShieldGemma models were fine-tuned on synthetically generated internal data and
+publicly available datasets. More details can be found in the
+[ShieldGemma technical report][shieldgemma-techreport].
+
+## Implementation Information
+
+### Hardware
+
+ShieldGemma was trained using the latest generation of
+[Tensor Processing Unit (TPU)][tpu] hardware (TPUv5e), for more details refer to
+the [Gemma 2 model card][gemma2-model-card].
+
+### Software
+
+Training was done using [JAX][jax] and [ML Pathways][ml-pathways]. For more
+details refer to the [Gemma 2 model card][gemma2-model-card].
+
+## Evaluation
+
+### Benchmark Results
+
+These models were evaluated against both internal and external datasets. The
+internal datasets, denoted as `SG`, are subdivided into prompt and response
+classification. Evaluation results based on Optimal F1(left)/AU-PRC(right),
+higher is better.
+
+| Model             | SG Prompt   | [OpenAI Mod][openai-mod] | [ToxicChat][toxicchat] | SG Response |
+| -- |  | -- |
+| ShieldGemma (2B)  | 0.825/0.887 | 0.812/0.887              | 0.704/0.778            | 0.743/0.802 |
+| ShieldGemma (9B)  | 0.828/0.894 | 0.821/0.907              | 0.694/0.782            | 0.753/0.817 |
+| ShieldGemma (27B) | 0.830/0.883 | 0.805/0.886              | 0.729/0.811            | 0.758/0.806 |
+| OpenAI Mod API    | 0.782/0.840 | 0.790/0.856              | 0.254/0.588            | -           |
+| LlamaGuard1 (7B)  | -           | 0.758/0.847              | 0.616/0.626            | -           |
+| LlamaGuard2 (8B)  | -           | 0.761/-                  | 0.471/-                | -           |
+| WildGuard (7B)    | 0.779/-     | 0.721/-                  | 0.708/-                | 0.656/-     |
+| GPT-4             | 0.810/0.847 | 0.705/-                  | 0.683/-                | 0.713/0.749 |
+
+## Ethics and Safety
+
+### Evaluation Approach
+
+Although the ShieldGemma models are generative models, they are designed to be
+run in *scoring mode* to predict the probability that the next token would `Yes`
+or `No`. Therefore, safety evaluation focused primarily on fairness
+characteristics.
+
+### Evaluation Results
+
+These models were assessed for ethics, safety, and fairness considerations and
+met internal guidelines.
+
+## Usage and Limitations
+
+These models have certain limitations that users should be aware of.
+
+### Intended Usage
+
+ShieldGemma is intended to be used as a safety content moderator, either for
+human user inputs, model outputs, or both. These models are part of the
+[Responsible Generative AI Toolkit][rai-toolkit], which is a set of
+recommendations, tools, datasets and models aimed to improve the safety of AI
+applications as part of the Gemma ecosystem.
+
+### Limitations
+
+All the usual limitations for large language models apply, see the
+[Gemma 2 model card][gemma2-model-card] for more details. Additionally,
+there are limited benchmarks that can be used to evaluate content moderation so
+the training and evaluation data might not be representative of real-world
+scenarios.
+
+ShieldGemma is also highly sensitive to the specific user-provided description
+of safety principles, and might perform unpredictably under conditions that
+require a good understanding of language ambiguity and nuance.
+
+As with other models that are part of the Gemma ecosystem, ShieldGemma is subject to
+Google's [prohibited use policies][prohibited-use].
+
+### Ethical Considerations and Risks
+
+The development of large language models (LLMs) raises several ethical concerns.
+We have carefully considered multiple aspects in the development of these
+models.
+
+Refer to the [Gemma model card][gemma2-model-card] for more details.
+
+### Benefits
+
+At the time of release, this family of models provides high-performance open
+large language model implementations designed from the ground up for Responsible
+AI development compared to similarly sized models.
+
+Using the benchmark evaluation metrics described in this document, these models
+have been shown to provide superior performance to other, comparably-sized open
+model alternatives.
+
+[rai-toolkit]: https://ai.google.dev/responsible
+[gemma2]: https://ai.google.dev/gemma#gemma-2
+[gemma2-model-card]: https://ai.google.dev/gemma/docs/model_card_2
+[shieldgemma]: https://ai.google.dev/gemma/docs/shieldgemma
+[shieldgemma-colab]: https://colab.research.google.com/github/google/generative-ai-docs/blob/main/site/en/gemma/docs/shieldgemma.ipynb
+[shieldgemma-kaggle]: https://www.kaggle.com/models/google/shieldgemma
+[shieldgemma-hfhub]: https://huggingface.co/models?search=shieldgemma
+[shieldgemma-techreport]: https://storage.googleapis.com/deepmind-media/gemma/shieldgemma-report.pdf
+[openai-mod]: https://github.com/openai/moderation-api-release
+[terms]: https://ai.google.dev/gemma/terms
+[toxicchat]: https://arxiv.org/abs/2310.17389
+[safety-policies]: https://storage.googleapis.com/gweb-uniblog-publish-prod/documents/2023_Google_AI_Principles_Progress_Update.pdf#page=11
+[prohibited-use]: https://ai.google.dev/gemma/prohibited_use_policy
+[tpu]: https://cloud.google.com/tpu/docs/intro-to-tpu
+[jax]: https://github.com/google/jax
+[ml-pathways]: https://blog.google/technology/ai/introducing-pathways-next-generation-ai-architecture/
+[llm-as-judge]: https://arxiv.org/abs/2306.05685
+
+<!-- original-model-card end -->