phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-12 21:58:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

from models.llama3_1 --> from llama_models.llama3_1

Browse source
This commit is contained in:
Hardik Shah 2024-07-21 19:07:02 -07:00
parent c6ef16f6bd
commit c64b8cba22
22 changed files with 29 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

2
toolchain/safety/shields/base.py
View file

@ -1,7 +1,7 @@
from abc import ABC, abstractmethod
from typing import List, Union
from models.llama3_1.api.datatypes import Attachment, Message
from llama_models.llama3_1.api.datatypes import Attachment, Message
from toolchain.safety.api.datatypes import * # noqa: F403
CANNED_RESPONSE_TEXT = "I can't answer that. Can I help with something else?"

2
toolchain/safety/shields/contrib/third_party_shield.py
View file

@ -1,7 +1,7 @@
import sys
from typing import List
from models.llama3_1.api.datatypes import Message
from llama_models.llama3_1.api.datatypes import Message
parent_dir = "../.."
sys.path.append(parent_dir)

2
toolchain/safety/shields/llama_guard.py
View file

@ -4,7 +4,7 @@ from string import Template
from typing import List, Optional
import torch
from models.llama3_1.api.datatypes import Message
from llama_models.llama3_1.api.datatypes import Message
from termcolor import cprint
from transformers import AutoModelForCausalLM, AutoTokenizer

2
toolchain/safety/shields/prompt_guard.py
View file

@ -3,7 +3,7 @@ from typing import List
import torch
from models.llama3_1.api.datatypes import Message
from llama_models.llama3_1.api.datatypes import Message
from termcolor import cprint
from transformers import AutoModelForSequenceClassification, AutoTokenizer

2
toolchain/safety/shields/shield_runner.py
View file

@ -1,7 +1,7 @@
import asyncio
from typing import List
from models.llama3_1.api.datatypes import Message, Role
from llama_models.llama3_1.api.datatypes import Message, Role
from .base import OnViolationAction, ShieldBase, ShieldResponse