Downloads · 30 days
507
1% of all-time downloads
ShieldLM-7B-internlm2
thu-coai/ShieldLM-7B-internlm2
ShieldLM-7B-internlm2 is a feature extraction model from thu-coai. Use it when you need embeddings to search or compare text. It is set up for transformers. The card lists the license as mit.
The ShieldLM model (paper link) initialized from internlm2-chat-7b. ShieldLM is a bilingual (Chinese and English) safety detector that mainly aims to help to detect safety issues in LLMs' generations. It aligns with g…
All-time downloads
44.6K
Public
Parameters
7.7B
15.5 GB on disk
Likes
11
Public
Repo makeup
Click a slice to open those files.
.safetensors15.5 GB · 100%
At a glance
- Task
- Feature Extraction
- Library
- transformers
- License
- mit
- Model type
- internlm2
- Access
- Public
- Created
- Feb 26, 2024
- Updated
- Feb 27, 2024
- SHA
- 92484e64
Model card
From the Hugging Face model README
Introduction
The ShieldLM model (paper link) initialized from internlm2-chat-7b. ShieldLM is a bilingual (Chinese and English) safety detector that mainly aims to help to detect safety issues in LLMs' generations. It aligns with general human safety standards, supports fine-grained customizable detection rules, and provides explanations for its decisions. Refer to our github repository for more detailed information.
Usage
Please refer to our github repository for the detailed usage instructions.
Performance
ShieldLM demonstrates impressive detection performance across 4 ID and OOD test sets, compared to strong baselines such as GPT-4, Llama Guard and Perspective API. Refer to our paper for more detailed evaluation results.
- Task
- Feature Extraction
- Library
- transformers
- Type
- internlm2
- License
- mit
- Languages
- en, zh
- Created
- Feb 26, 2024
- Updated
- Feb 27, 2024