adding vision guard to Together safety

This commit is contained in:
Yogish Baliga 2024-09-27 13:33:33 -07:00
parent d7c55f0ad0
commit 27b63f4de5
3 changed files with 2 additions and 5 deletions

View file

@ -19,6 +19,7 @@ from .config import TogetherSafetyConfig
SAFETY_SHIELD_TYPES = {
"Llama-Guard-3-8B": "meta-llama/Meta-Llama-Guard-3-8B",
"Llama-Guard-3-11B-Vision": "meta-llama/Llama-Guard-3-11B-Vision-Turbo",
}