phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 15:23:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

adding vision guard to Together safety

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/adapters/inference/together/__init__.py
View file

@ -4,7 +4,7 @@
# This source code is licensed under the terms described in the LICENSE file in # This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree. # the root directory of this source tree.
from .config import TogetherImplConfig, TogetherHeaderExtractor from .config import TogetherImplConfig
async def get_adapter_impl(config: TogetherImplConfig, _deps): async def get_adapter_impl(config: TogetherImplConfig, _deps):

4
llama_stack/providers/adapters/inference/together/config.py
View file

@ -8,10 +8,6 @@ from llama_models.schema_utils import json_schema_type
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
class TogetherHeaderExtractor(BaseModel):
together_api_key: str
@json_schema_type @json_schema_type
class TogetherImplConfig(BaseModel): class TogetherImplConfig(BaseModel):
url: str = Field( url: str = Field(

1
llama_stack/providers/adapters/safety/together/together.py
View file

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