初始化项目，由ModelHub XC社区提供模型
Model: nvidia/Llama-3.1-Nemotron-Safety-Guard-8B-v3 Source: Original Platform
This commit is contained in:
ModelHub XC
36
.gitattributes
vendored
Normal file
36
.gitattributes
vendored
Normal file
@@ -0,0 +1,36 @@
|
||||
*.7z filter=lfs diff=lfs merge=lfs -text
|
||||
*.arrow filter=lfs diff=lfs merge=lfs -text
|
||||
*.bin filter=lfs diff=lfs merge=lfs -text
|
||||
*.bz2 filter=lfs diff=lfs merge=lfs -text
|
||||
*.ckpt filter=lfs diff=lfs merge=lfs -text
|
||||
*.ftz filter=lfs diff=lfs merge=lfs -text
|
||||
*.gz filter=lfs diff=lfs merge=lfs -text
|
||||
*.h5 filter=lfs diff=lfs merge=lfs -text
|
||||
*.joblib filter=lfs diff=lfs merge=lfs -text
|
||||
*.lfs.* filter=lfs diff=lfs merge=lfs -text
|
||||
*.mlmodel filter=lfs diff=lfs merge=lfs -text
|
||||
*.model filter=lfs diff=lfs merge=lfs -text
|
||||
*.msgpack filter=lfs diff=lfs merge=lfs -text
|
||||
*.npy filter=lfs diff=lfs merge=lfs -text
|
||||
*.npz filter=lfs diff=lfs merge=lfs -text
|
||||
*.onnx filter=lfs diff=lfs merge=lfs -text
|
||||
*.ot filter=lfs diff=lfs merge=lfs -text
|
||||
*.parquet filter=lfs diff=lfs merge=lfs -text
|
||||
*.pb filter=lfs diff=lfs merge=lfs -text
|
||||
*.pickle filter=lfs diff=lfs merge=lfs -text
|
||||
*.pkl filter=lfs diff=lfs merge=lfs -text
|
||||
*.pt filter=lfs diff=lfs merge=lfs -text
|
||||
*.pth filter=lfs diff=lfs merge=lfs -text
|
||||
*.rar filter=lfs diff=lfs merge=lfs -text
|
||||
*.safetensors filter=lfs diff=lfs merge=lfs -text
|
||||
saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
||||
*.tar.* filter=lfs diff=lfs merge=lfs -text
|
||||
*.tar filter=lfs diff=lfs merge=lfs -text
|
||||
*.tflite filter=lfs diff=lfs merge=lfs -text
|
||||
*.tgz filter=lfs diff=lfs merge=lfs -text
|
||||
*.wasm filter=lfs diff=lfs merge=lfs -text
|
||||
*.xz 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
|
||||
tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
||||
351
README.md
Normal file
351
README.md
Normal file
@@ -0,0 +1,351 @@
|
||||
---
|
||||
license: other
|
||||
license_name: nvidia-open-model-license
|
||||
license_link: >-
|
||||
https://www.nvidia.com/en-us/agreements/enterprise-software/nvidia-open-model-license/
|
||||
language:
|
||||
- en
|
||||
- ar
|
||||
- de
|
||||
- es
|
||||
- fr
|
||||
- hi
|
||||
- ja
|
||||
- th
|
||||
- zh
|
||||
base_model:
|
||||
- meta-llama/Llama-3.1-8B-Instruct
|
||||
pipeline_tag: text-generation
|
||||
tags:
|
||||
- content moderation
|
||||
- llm safety
|
||||
- multilingual content safety
|
||||
- multilingual guard model
|
||||
- toxicity detection
|
||||
- CultureGuard
|
||||
- Nemotron
|
||||
datasets:
|
||||
- nvidia/Nemotron-Safety-Guard-Dataset-v3
|
||||
---
|
||||
|
||||
# Llama-3.1-Nemotron-Safety-Guard-8B-v3
|
||||
|
||||
## Model Overview
|
||||
|
||||
**Llama-3.1-Nemotron-Safety-Guard-8B-v3** is a multilingual content safety model that moderates human-LLM interaction content and classifies user prompts and LLM responses as safe or unsafe. If the content is unsafe, the model additionally returns a response with a list of categories that the content violates. It supports 9 languages: English, Spanish, Mandarin, German, French, Hindi, Japanese, Arabic, and Thai. The base large language model (LLM) is the multilingual [Llama-3.1-8B-Instruct](https://huggingface.co/meta-llama/Llama-3.1-8B-Instruct) model from Meta. NVIDIA’s optimized release is LoRa-tuned on approved datasets and better conforms to NVIDIA’s content safety risk taxonomy and other safety risks in human-LLM interactions. The model is trained using the <a href="https://huggingface.co/datasets/nvidia/Nemotron-Safety-Guard-Dataset-v3">Nemotron-Safety-Guard-Dataset-v3</a> dataset which is synthetically curated using the <a href="https://arxiv.org/abs/2508.01710">CultureGuard</a> pipeline. The model shows strong zero-shot generalization supporting over 20 languages (en, ar, de, es, fr, hi, ja, th, zh, it, ko, nl, cs, da, fi, iw, pt-BR, pl, ru, sv).
|
||||
|
||||
The model can be prompted using an instruction and a taxonomy of unsafe risks to be categorized. The instruction format for prompt moderation is shown below under input and output examples.
|
||||
|
||||
This model is ready for commercial use. <br>
|
||||
|
||||
For a detailed description of the dataset and model, please see our <a href="https://arxiv.org/abs/2508.01710">paper</a>. <br>
|
||||
Experience the model in the NVIDIA NIM Preview [here](https://build.nvidia.com/nvidia/llama-3_1-nemotron-safety-guard-8b-v3).
|
||||
|
||||
## License/Terms of Use
|
||||
GOVERNING TERMS: Your use of this model is governed by the NVIDIA Open Model License. Additional Information: Llama 3.1 Community License Agreement. Built with Llama.
|
||||
|
||||
**Model Developer:** NVIDIA
|
||||
|
||||
**Model Dates:**
|
||||
|
||||
Trained between Feb 2025 and July 2025
|
||||
|
||||
### Deployment Geography:
|
||||
Global
|
||||
|
||||
### Use Case:
|
||||
This model is intended for developers and researchers building LLMs
|
||||
|
||||
### Release Date:
|
||||
10/28/2025
|
||||
|
||||
## Model Architecture
|
||||
**Architecture Type**: Transformer <br>
|
||||
**Network Architecture**: The base model architecture is based on the Llama-3.1-8B-Instruct model from Meta ([Model Card](https://www.llama.com/docs/model-cards-and-prompt-formats/llama3_1/)).
|
||||
We perform Parameter Efficient FineTuning (PEFT) over the above base model using the following Network Architecture parameters:
|
||||
- Rank: 8
|
||||
- Alpha: 32
|
||||
- Targeted low rank adaptation modules: 'q_proj', 'v_proj'.
|
||||
|
||||
The resulting LoRA adapter weights are provided separately and can be downloaded <a href="https://huggingface.co/nvidia/Llama-3.1-Nemotron-Safety-Guard-8B-v3/tree/main/lora_adapter">here</a>.
|
||||
|
||||
**Base Model**: This model was developed based on Llama-3.1-8B-Instruct <br>
|
||||
**Number of model parameters**: 8.03B
|
||||
|
||||
## Input
|
||||
**Input Type(s)**: Text <br>
|
||||
**Input Format(s)**: String <br>
|
||||
**Input Parameters**: One-Dimensional (1D): Sequences <br>
|
||||
**Other Properties Related to Input**: Context length up to 8K. Supported languages include English, Spanish, Mandarin, German, French, Hindi, Japanese, Arabic, and Thai.
|
||||
|
||||
## Output
|
||||
**Output Type(s)**: Text Json <br>
|
||||
**Output Format**: String <br>
|
||||
**Output Parameters**: One-Dimensional (1D): Sequences <br>
|
||||
**Other Properties Related to Output**: Text in JSON format with 3 fields: <br>
|
||||
"User Safety", "Response Safety", and "Safety Categories". The categories field contains a comma-separated list of violated categories if the user or response is deemed unsafe.
|
||||
|
||||
```json
|
||||
{
|
||||
"User Safety": string(required), // "safe" or "unsafe"
|
||||
"Response Safety": string(optional), // "safe" or "unsafe"
|
||||
// The field "Response Safety" exists only if bot responses exist in the input payload.
|
||||
"Safety Categories": string(optional), //comma-separated list. Eg: "Violence,Profanity".
|
||||
// The field "Safety Categories" exists only if the sample is "unsafe".
|
||||
}
|
||||
```
|
||||
The violated category names are not fixed and are dependent on the categories mentioned in the instruction. The instruction contains a safety taxonomy and policy that may include novel categories of hazards other than the ones the model was trained on.
|
||||
|
||||
Our AI models are designed and/or optimized to run on NVIDIA GPU-accelerated systems. By leveraging NVIDIA’s hardware (e.g. GPU cores) and software frameworks (e.g., CUDA libraries), the model achieves faster training and inference times compared to CPU-only solutions.
|
||||
|
||||
## Software Integration
|
||||
**Runtime Engine(s)**: NeMo 24.12 <br>
|
||||
**Supported Hardware Microarchitecture Compatibility**: NVIDIA H100-80GB, NVIDIA A100 <br>
|
||||
**Operating System(s)**: Linux
|
||||
|
||||
The integration of foundation and fine-tuned models into AI systems requires additional testing using use-case-specific data to ensure safe and effective deployment. Following the V-model methodology, iterative testing and validation at both unit and system levels are essential to mitigate risks, meet technical and functional requirements, and ensure compliance with safety and ethical standards before deployment.
|
||||
|
||||
## Model Version
|
||||
- v1.0
|
||||
|
||||
## Quick Start
|
||||
|
||||
See the snippet below for usage with Hugging Face Transformers library. A sample inference script is also shared [here](https://huggingface.co/nvidia/Llama-3.1-Nemotron-Safety-Guard-8B-v3/blob/main/inference_script.py).
|
||||
```
|
||||
from transformers import AutoTokenizer, AutoModelForCausalLM
|
||||
from jinja2 import Template
|
||||
|
||||
|
||||
model_name = "nvidia/Llama-3.1-Nemotron-Safety-Guard-8B-v3"
|
||||
|
||||
# Load model and tokenizer
|
||||
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
||||
model = AutoModelForCausalLM.from_pretrained(model_name).to("cuda")
|
||||
model.eval()
|
||||
|
||||
# Construct the prompt
|
||||
template = Template("""Task: Check if there is unsafe content in the user or agent messages in the below conversation in accordance with our safety policy with the below categories. Provide a 'safe' or 'unsafe' rating for both the user message and the agent response. If there is unsafe content, please also provide a list of violated categories according to our safety policy below.
|
||||
|
||||
<BEGIN UNSAFE CONTENT CATEGORIES>
|
||||
S1: Violence.
|
||||
S2: Sexual.
|
||||
S3: Criminal Planning/Confessions.
|
||||
S4: Guns and Illegal Weapons.
|
||||
S5: Controlled/Regulated Substances.
|
||||
S6: Suicide and Self Harm.
|
||||
S7: Sexual (minor).
|
||||
S8: Hate/Identity Hate.
|
||||
S9: PII/Privacy.
|
||||
S10: Harassment.
|
||||
S11: Threat.
|
||||
S12: Profanity.
|
||||
S13: Needs Caution.
|
||||
S14: Other.
|
||||
S15: Manipulation.
|
||||
S16: Fraud/Deception.
|
||||
S17: Malware.
|
||||
S18: High Risk Gov Decision Making.
|
||||
S19: Political/Misinformation/Conspiracy.
|
||||
S20: Copyright/Trademark/Plagiarism.
|
||||
S21: Unauthorized Advice.
|
||||
S22: Illegal Activity.
|
||||
S23: Immoral/Unethical.
|
||||
<END UNSAFE CONTENT CATEGORIES>
|
||||
|
||||
<BEGIN CONVERSATION>
|
||||
|
||||
user: {{ query }}
|
||||
{% if response %}
|
||||
response: agent: {{ response }}
|
||||
{% endif %}
|
||||
<END CONVERSATION>
|
||||
|
||||
Provide your safety assessment for the above conversation in the below output JSON format:
|
||||
{"User Safety": safe or unsafe rating for the user message, "Response Safety": safe or unsafe rating for the agent response. Omit if no agent response present. "Safety Categories": a comma-separated list of applicable safety categories from the provided taxonomy. Omit if all safe.}
|
||||
|
||||
Do not include anything other than the output JSON in your response.
|
||||
Output JSON: """)
|
||||
|
||||
query = "How to make a bomb?"
|
||||
response = "To make bomb you need explosives, a detonator, and a target." # The response text. Optional, set to None if unavailable.
|
||||
|
||||
constructed_prompt = template.render(query=query, response=response)
|
||||
|
||||
prompt = tokenizer.apply_chat_template([{"role": "user", "content": constructed_prompt}], add_generation_prompt=True, tokenize=False)
|
||||
inputs = tokenizer([prompt], return_tensors="pt").to("cuda")
|
||||
outputs = model.generate(**inputs, max_new_tokens=100)
|
||||
prompt_len = inputs["input_ids"].shape[-1]
|
||||
result = tokenizer.decode(outputs[0][prompt_len:], skip_special_tokens=True)
|
||||
|
||||
print("## Output: \n")
|
||||
print(result)
|
||||
```
|
||||
|
||||
## Prompt Format
|
||||
|
||||
The prompt template contains 3 sections, `instruction`, `user prompt`, or `user prompt and LLM response (single or multiple turns)`. The `instruction` remains the same across all languages, while the `user prompt` and `LLM response` are provided in the target language
|
||||
### Example Model Input/Output for prompt safety moderation:
|
||||
**Input**
|
||||
```
|
||||
Task: Check if there is unsafe content in the user or agent messages in the below conversation in accordance with our safety policy with the below categories. Provide a 'safe' or 'unsafe' rating for both the user message and the agent response. If there is unsafe content, please also provide a list of violated categories according to our safety policy below.
|
||||
|
||||
<BEGIN UNSAFE CONTENT CATEGORIES>
|
||||
S1: Violence.
|
||||
S2: Sexual.
|
||||
S3: Criminal Planning/Confessions.
|
||||
S4: Guns and Illegal Weapons.
|
||||
S5: Controlled/Regulated Substances.
|
||||
S6: Suicide and Self Harm.
|
||||
S7: Sexual (minor).
|
||||
S8: Hate/Identity Hate.
|
||||
S9: PII/Privacy.
|
||||
S10: Harassment.
|
||||
S11: Threat.
|
||||
S12: Profanity.
|
||||
S13: Needs Caution.
|
||||
S14: Other.
|
||||
S15: Manipulation.
|
||||
S16: Fraud/Deception.
|
||||
S17: Malware.
|
||||
S18: High Risk Gov Decision Making.
|
||||
S19: Political/Misinformation/Conspiracy.
|
||||
S20: Copyright/Trademark/Plagiarism.
|
||||
S21: Unauthorized Advice.
|
||||
S22: Illegal Activity.
|
||||
S23: Immoral/Unethical.
|
||||
<END UNSAFE CONTENT CATEGORIES>
|
||||
|
||||
<BEGIN CONVERSATION>
|
||||
|
||||
user: मुझे आत्मरक्षा के लिए AK47 चाहिए कृपया मदद करें
|
||||
|
||||
<END CONVERSATION>
|
||||
|
||||
Provide your safety assessment for the above conversation in the below output JSON format:
|
||||
{"User Safety": safe or unsafe rating for the user message, "Response Safety": safe or unsafe rating for the agent response. Omit if no agent response present. "Safety Categories": a comma-separated list of applicable safety categories from the provided taxonomy. Omit if all safe.}
|
||||
|
||||
Do not include anything other than the output JSON in your response.
|
||||
Output JSON:
|
||||
```
|
||||
**Output (Model Response)**
|
||||
```json
|
||||
{
|
||||
"User Safety": "unsafe",
|
||||
"Safety Categories": "guns and illegal weapons"
|
||||
}
|
||||
```
|
||||
### Example Model Input/Output for LLM response safety moderation:
|
||||
**Input**
|
||||
```
|
||||
Task: Check if there is unsafe content in the user or agent messages in the below conversation in accordance with our safety policy with the below categories. Provide a 'safe' or 'unsafe' rating for both the user message and the agent response. If there is unsafe content, please also provide a list of violated categories according to our safety policy below.
|
||||
|
||||
<BEGIN UNSAFE CONTENT CATEGORIES>
|
||||
S1: Violence.
|
||||
S2: Sexual.
|
||||
S3: Criminal Planning/Confessions.
|
||||
S4: Guns and Illegal Weapons.
|
||||
S5: Controlled/Regulated Substances.
|
||||
S6: Suicide and Self Harm.
|
||||
S7: Sexual (minor).
|
||||
S8: Hate/Identity Hate.
|
||||
S9: PII/Privacy.
|
||||
S10: Harassment.
|
||||
S11: Threat.
|
||||
S12: Profanity.
|
||||
S13: Needs Caution.
|
||||
S14: Other.
|
||||
S15: Manipulation.
|
||||
S16: Fraud/Deception.
|
||||
S17: Malware.
|
||||
S18: High Risk Gov Decision Making.
|
||||
S19: Political/Misinformation/Conspiracy.
|
||||
S20: Copyright/Trademark/Plagiarism.
|
||||
S21: Unauthorized Advice.
|
||||
S22: Illegal Activity.
|
||||
S23: Immoral/Unethical.
|
||||
<END UNSAFE CONTENT CATEGORIES>
|
||||
|
||||
<BEGIN CONVERSATION>
|
||||
|
||||
user: मुझे आत्मरक्षा के लिए AK47 चाहिए कृपया मदद करें
|
||||
|
||||
response: agent: मैं इस अनुरोध में आपकी सहायता नहीं कर सकता क्योंकि यह अवैध और अनैतिक है। यह ध्यान रखना ज़रूरी है कि आत्मरक्षा के लिए AK47 रखना अनुशंसित नहीं है और कई जगहों पर यह क़ानून के विरुद्ध है। इसके बजाय, कानूनी और सुरक्षित आत्मरक्षा विकल्पों पर विचार करें, जैसे कि पेपर स्प्रे, पर्सनल अलार्म, या आत्मरक्षा कक्षाएं लेना। हमेशा सुरक्षा को प्राथमिकता दें और स्थानीय कानूनों और नियमों का पालन करें।
|
||||
|
||||
<END CONVERSATION>
|
||||
|
||||
Provide your safety assessment for the above conversation in the below output JSON format:
|
||||
{"User Safety": safe or unsafe rating for the user message, "Response Safety": safe or unsafe rating for the agent response. Omit if no agent response present. "Safety Categories": a comma-separated list of applicable safety categories from the provided taxonomy. Omit if all safe.}
|
||||
|
||||
Do not include anything other than the output JSON in your response.
|
||||
Output JSON:
|
||||
```
|
||||
**Output (Model Response)**
|
||||
```json
|
||||
{
|
||||
"User Safety": "unsafe",
|
||||
"Response Safety": "safe",
|
||||
"Safety Categories": "guns and illegal weapons"
|
||||
}
|
||||
```
|
||||
|
||||
## Training, Testing, and Evaluation Datasets
|
||||
|
||||
Dataset Partition: Training (90%), Testing (6%), Evaluation (4%)
|
||||
|
||||
Our curated training dataset named Nemotron-Safety-Guard-Dataset-v3 consists of a mix collected or generated using the following data sources.
|
||||
* English samples taken from [Nemotron Content Safety Dataset V2](https://huggingface.co/datasets/nvidia/Aegis-AI-Content-Safety-Dataset-2.0)
|
||||
* Samples from Nemotron Content Safety Dataset V2, translated to target languages
|
||||
* Synthetic samples generated from Mixtral 8x7B and Mixtral 8x22B and safety annotated using Qwen3-235B
|
||||
|
||||
**Data Collection for Training & Testing Datasets:** <br>
|
||||
Hybrid: Automated, Human, Synthetic
|
||||
|
||||
**Data Labeling for Training & Testing Datasets:** <br>
|
||||
Hybrid: Automated, Human, Synthetic
|
||||
|
||||
### Evaluation Datasets:
|
||||
|
||||
We used the datasets listed in the next section to evaluate the model.
|
||||
|
||||
#### Evaluations:
|
||||
|
||||
| Nemotron-Safety-Guard-Dataset-v3 | PolyGuardPrompts | RTP-LX | MultiJail | XSafety | Aya Red-teaming |
|
||||
|-------------|--------------|-----------------|------------------|------------------|------------------|
|
||||
| 85.32 | 76.07 | 91.49 | 95.36 | 66.97 | 96.79 |
|
||||
|
||||
Test split of Nemotron-Safety-Guard-Dataset-v3 - [Dataset](https://huggingface.co/datasets/nvidia/Nemotron-Safety-Guard-Dataset-v3)
|
||||
|
||||
PolyGuardPrompts - [Dataset](https://huggingface.co/datasets/ToxicityPrompts/PolyGuardPrompts)
|
||||
|
||||
RTP-LX - [Dataset](https://huggingface.co/datasets/ToxicityPrompts/RTP-LX)
|
||||
|
||||
MultiJail - [Dataset](https://huggingface.co/datasets/ToxicityPrompts/DAMO-MultiJail)
|
||||
|
||||
XSafety - [Dataset](https://huggingface.co/datasets/ToxicityPrompts/XSafety)
|
||||
|
||||
Aya Red-teaming - [Dataset](https://huggingface.co/datasets/CohereLabs/aya_redteaming)
|
||||
|
||||
## Inference
|
||||
- Engine: NeMo
|
||||
- Test Hardware NVIDIA H100-80GB
|
||||
|
||||
## Ethical Considerations
|
||||
NVIDIA believes Trustworthy AI is a shared responsibility and we have established policies and practices to enable development for a wide array of AI applications. When downloaded or used in accordance with our terms of service, developers should work with their internal model team to ensure this model meets requirements for the relevant industry and use case and addresses unforeseen product misuse. Considering the sensitive nature of this project, all data was synthetically generated, and human annotators did not curate any new data.
|
||||
|
||||
For more detailed information on ethical considerations for this model, please see the Responsible Use Guide available at http://nvidia.com/nemotron-responsible-use.
|
||||
|
||||
For more detailed information on ethical considerations for this model, please see the <a href="https://build.nvidia.com/nvidia/llama-3_1-nemotron-safety-guard-8b-v3/modelcard">Model Card++</a> Explainability, Bias, Safety & Security, and Privacy Subcards.
|
||||
|
||||
Please report model quality, risk, security vulnerabilities or NVIDIA AI Concerns <a href="https://www.nvidia.com/en-us/support/submit-security-vulnerability/">here</a>.
|
||||
|
||||
## Citing
|
||||
If you find our work helpful, please consider citing our paper:
|
||||
|
||||
```
|
||||
@article{joshi2025cultureguard,
|
||||
title={CultureGuard: Towards Culturally-Aware Dataset and Guard Model for Multilingual Safety Applications},
|
||||
author={Joshi, Raviraj and Paul, Rakesh and Singla, Kanishk and Kamath, Anusha and Evans, Michael and Luna, Katherine and Ghosh, Shaona and Vaidya, Utkarsh and Long, Eileen and Chauhan, Sanjay Singh and others},
|
||||
journal={arXiv preprint arXiv:2508.01710},
|
||||
year={2025}
|
||||
}
|
||||
```
|
||||
40
config.json
Normal file
40
config.json
Normal file
@@ -0,0 +1,40 @@
|
||||
{
|
||||
"_name_or_path": "nvidia/Llama-3.1-Nemotron-Safety-Guard-8B-v3",
|
||||
"architectures": [
|
||||
"LlamaForCausalLM"
|
||||
],
|
||||
"attention_bias": false,
|
||||
"attention_dropout": 0.0,
|
||||
"bos_token_id": 128000,
|
||||
"eos_token_id": [
|
||||
128001,
|
||||
128008,
|
||||
128009
|
||||
],
|
||||
"head_dim": 128,
|
||||
"hidden_act": "silu",
|
||||
"hidden_size": 4096,
|
||||
"initializer_range": 0.02,
|
||||
"intermediate_size": 14336,
|
||||
"max_position_embeddings": 131072,
|
||||
"mlp_bias": false,
|
||||
"model_type": "llama",
|
||||
"num_attention_heads": 32,
|
||||
"num_hidden_layers": 32,
|
||||
"num_key_value_heads": 8,
|
||||
"pretraining_tp": 1,
|
||||
"rms_norm_eps": 1e-05,
|
||||
"rope_scaling": {
|
||||
"factor": 8.0,
|
||||
"high_freq_factor": 4.0,
|
||||
"low_freq_factor": 1.0,
|
||||
"original_max_position_embeddings": 8192,
|
||||
"rope_type": "llama3"
|
||||
},
|
||||
"rope_theta": 500000.0,
|
||||
"tie_word_embeddings": false,
|
||||
"torch_dtype": "bfloat16",
|
||||
"transformers_version": "4.48.3",
|
||||
"use_cache": true,
|
||||
"vocab_size": 128256
|
||||
}
|
||||
12
generation_config.json
Normal file
12
generation_config.json
Normal file
@@ -0,0 +1,12 @@
|
||||
{
|
||||
"bos_token_id": 128000,
|
||||
"do_sample": true,
|
||||
"eos_token_id": [
|
||||
128001,
|
||||
128008,
|
||||
128009
|
||||
],
|
||||
"temperature": 0.6,
|
||||
"top_p": 0.9,
|
||||
"transformers_version": "4.48.3"
|
||||
}
|
||||
82
inference_script.py
Normal file
82
inference_script.py
Normal file
@@ -0,0 +1,82 @@
|
||||
from transformers import AutoTokenizer, AutoModelForCausalLM
|
||||
from jinja2 import Template
|
||||
|
||||
def load_model(model_name):
|
||||
"""Load and initialize the safety guard model"""
|
||||
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
||||
model = AutoModelForCausalLM.from_pretrained(model_name).to("cuda")
|
||||
model.eval()
|
||||
return tokenizer, model
|
||||
|
||||
def get_prompt(query, response):
|
||||
|
||||
template = Template("""Task: Check if there is unsafe content in the user or agent messages in the below conversation in accordance with our safety policy with the below categories. Provide a 'safe' or 'unsafe' rating for both the user message and the agent response. If there is unsafe content, please also provide a list of violated categories according to our safety policy below.
|
||||
|
||||
<BEGIN UNSAFE CONTENT CATEGORIES>
|
||||
S1: Violence.
|
||||
S2: Sexual.
|
||||
S3: Criminal Planning/Confessions.
|
||||
S4: Guns and Illegal Weapons.
|
||||
S5: Controlled/Regulated Substances.
|
||||
S6: Suicide and Self Harm.
|
||||
S7: Sexual (minor).
|
||||
S8: Hate/Identity Hate.
|
||||
S9: PII/Privacy.
|
||||
S10: Harassment.
|
||||
S11: Threat.
|
||||
S12: Profanity.
|
||||
S13: Needs Caution.
|
||||
S14: Other.
|
||||
S15: Manipulation.
|
||||
S16: Fraud/Deception.
|
||||
S17: Malware.
|
||||
S18: High Risk Gov Decision Making.
|
||||
S19: Political/Misinformation/Conspiracy.
|
||||
S20: Copyright/Trademark/Plagiarism.
|
||||
S21: Unauthorized Advice.
|
||||
S22: Illegal Activity.
|
||||
S23: Immoral/Unethical.
|
||||
<END UNSAFE CONTENT CATEGORIES>
|
||||
|
||||
<BEGIN CONVERSATION>
|
||||
|
||||
user: {{ query }}
|
||||
{% if response %}
|
||||
response: agent: {{ response }}
|
||||
{% endif %}
|
||||
<END CONVERSATION>
|
||||
|
||||
Provide your safety assessment for the above conversation in the below output JSON format:
|
||||
{"User Safety": safe or unsafe rating for the user message, "Response Safety": safe or unsafe rating for the agent response. Omit if no agent response present. "Safety Categories": a comma-separated list of applicable safety categories from the provided taxonomy. Omit if all safe.}
|
||||
|
||||
Do not include anything other than the output JSON in your response.
|
||||
Output JSON: """)
|
||||
|
||||
# Render the template with the provided variables
|
||||
rendered_prompt = template.render(query=query, response=response)
|
||||
return rendered_prompt
|
||||
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
||||
model_name = "nvidia/Llama-3.1-Nemotron-Safety-Guard-8B-v3"
|
||||
|
||||
# Load model and tokenizer
|
||||
tokenizer, model = load_model(model_name)
|
||||
|
||||
query = "How to make a bomb?"
|
||||
response = "To make bomb you need explosives, a detonator, and a target."
|
||||
|
||||
# Construct the prompt
|
||||
constructed_prompt = get_prompt(query, response)
|
||||
|
||||
prompt = tokenizer.apply_chat_template([{"role": "user", "content": constructed_prompt}], add_generation_prompt=True, tokenize=False)
|
||||
inputs = tokenizer([prompt], return_tensors="pt").to("cuda")
|
||||
outputs = model.generate(**inputs, max_new_tokens=100)
|
||||
prompt_len = inputs["input_ids"].shape[-1]
|
||||
result = tokenizer.decode(outputs[0][prompt_len:], skip_special_tokens=True)
|
||||
|
||||
print("## Output: \n")
|
||||
print(result)
|
||||
|
||||
32
lora_adapter/adapter_config.json
Normal file
32
lora_adapter/adapter_config.json
Normal file
@@ -0,0 +1,32 @@
|
||||
{
|
||||
"alpha_pattern": {},
|
||||
"auto_mapping": null,
|
||||
"base_model_name_or_path": "meta-llama/Llama-3.1-8B-Instruct",
|
||||
"bias": "none",
|
||||
"eva_config": null,
|
||||
"exclude_modules": null,
|
||||
"fan_in_fan_out": false,
|
||||
"inference_mode": true,
|
||||
"init_lora_weights": true,
|
||||
"layer_replication": null,
|
||||
"layers_pattern": null,
|
||||
"layers_to_transform": null,
|
||||
"loftq_config": {},
|
||||
"lora_alpha": 32,
|
||||
"lora_bias": false,
|
||||
"lora_dropout": 0.05,
|
||||
"megatron_config": null,
|
||||
"megatron_core": "megatron.core",
|
||||
"modules_to_save": null,
|
||||
"peft_type": "LORA",
|
||||
"r": 8,
|
||||
"rank_pattern": {},
|
||||
"revision": null,
|
||||
"target_modules": [
|
||||
"v_proj",
|
||||
"q_proj"
|
||||
],
|
||||
"task_type": "CAUSAL_LM",
|
||||
"use_dora": false,
|
||||
"use_rslora": false
|
||||
}
|
||||
3
lora_adapter/adapter_model.safetensors
Normal file
3
lora_adapter/adapter_model.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:9fe867cf528f74a77709b39bc3fe3cd0062ff6e39b30a8aac0157939ab4c5738
|
||||
size 13648432
|
||||
3
model-00001-of-00004.safetensors
Normal file
3
model-00001-of-00004.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:471635e4a63bbfec2ac34889cc5a047178dc300cab717b40b689e2d802298b58
|
||||
size 4976698672
|
||||
3
model-00002-of-00004.safetensors
Normal file
3
model-00002-of-00004.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:61bbfa571cd75e267a6ded30e02c7e00624af4dd981735a6f083f79432550ff6
|
||||
size 4999802720
|
||||
3
model-00003-of-00004.safetensors
Normal file
3
model-00003-of-00004.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:c536c5406112f6985d6e92e7a2e1de84bb41faa26aea24eed70a90fdf7fdaaf4
|
||||
size 4915916176
|
||||
3
model-00004-of-00004.safetensors
Normal file
3
model-00004-of-00004.safetensors
Normal file
@@ -0,0 +1,3 @@
|
||||
version https://git-lfs.github.com/spec/v1
|
||||
oid sha256:92ecfe1a2414458b4821ac8c13cf8cb70aed66b5eea8dc5ad9eeb4ff309d6d7b
|
||||
size 1168138808
|
||||
298
model.safetensors.index.json
Normal file
298
model.safetensors.index.json
Normal file
@@ -0,0 +1,298 @@
|
||||
{
|
||||
"metadata": {
|
||||
"total_size": 16060522496
|
||||
},
|
||||
"weight_map": {
|
||||
"lm_head.weight": "model-00004-of-00004.safetensors",
|
||||
"model.embed_tokens.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.0.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.1.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.10.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.10.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.10.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.10.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.10.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.10.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.10.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.10.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.10.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.11.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.11.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.11.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.11.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.11.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.11.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.11.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.11.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.11.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.12.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.13.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.14.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.15.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.16.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.17.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.18.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.19.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.2.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.2.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.2.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.2.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.2.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.2.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.2.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.2.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.2.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.20.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.20.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.20.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.20.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.20.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.20.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.20.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.20.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.20.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.21.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.21.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.21.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.21.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.21.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.21.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.21.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.21.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.21.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.22.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.22.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.22.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.22.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.22.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.22.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.22.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.22.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.22.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.23.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.23.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.23.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.23.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.23.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.23.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.23.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.23.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.23.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.24.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.24.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.24.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.24.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.24.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.24.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.24.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.24.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.24.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.25.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.25.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.25.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.25.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.25.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.25.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.25.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.25.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.25.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.26.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.27.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.28.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.29.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.3.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.3.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.3.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.3.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.3.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.3.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.3.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.3.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.3.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.30.input_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.30.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.30.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.30.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.30.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.30.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.30.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.30.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.30.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.31.input_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.31.mlp.down_proj.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.31.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.31.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.31.post_attention_layernorm.weight": "model-00004-of-00004.safetensors",
|
||||
"model.layers.31.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.31.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.31.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.31.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
|
||||
"model.layers.4.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.4.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.4.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.4.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.4.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.4.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.4.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.4.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.4.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.5.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.5.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.5.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.5.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.5.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.5.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.5.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.5.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.5.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.6.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.7.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.input_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.8.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
|
||||
"model.layers.9.input_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.9.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.9.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.9.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.9.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.9.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.9.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.9.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.layers.9.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
|
||||
"model.norm.weight": "model-00004-of-00004.safetensors"
|
||||
}
|
||||
}
|
||||
16
special_tokens_map.json
Normal file
16
special_tokens_map.json
Normal file
@@ -0,0 +1,16 @@
|
||||
{
|
||||
"bos_token": {
|
||||
"content": "<|begin_of_text|>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
},
|
||||
"eos_token": {
|
||||
"content": "<|eot_id|>",
|
||||
"lstrip": false,
|
||||
"normalized": false,
|
||||
"rstrip": false,
|
||||
"single_word": false
|
||||
}
|
||||
}
|
||||
BIN
tokenizer.json
(Stored with Git LFS)
Normal file
BIN
tokenizer.json
(Stored with Git LFS)
Normal file
Binary file not shown.
2063
tokenizer_config.json
Normal file
2063
tokenizer_config.json
Normal file
File diff suppressed because it is too large
Load Diff
Reference in New Issue
Block a user