phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

llama_models.llama3_1 -> llama_models.llama3

Browse source
This commit is contained in:
Ashwin Bharambe 2024-08-19 10:55:37 -07:00
parent f502716cf7
commit 38244c3161
27 changed files with 44 additions and 42 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_toolchain/safety/api/datatypes.py
View file

@ -7,7 +7,7 @@
from enum import Enum
from typing import Dict, Optional, Union
from llama_models.llama3_1.api.datatypes import ToolParamDefinition
from llama_models.llama3.api.datatypes import ToolParamDefinition
from llama_models.schema_utils import json_schema_type
from pydantic import BaseModel, validator

2
llama_toolchain/safety/api/endpoints.py
View file

@ -7,7 +7,7 @@
from .datatypes import * # noqa: F403
from typing import List, Protocol
from llama_models.llama3_1.api.datatypes import Message
from llama_models.llama3.api.datatypes import Message
# this dependency is annoying and we need a forked up version anyway
from llama_models.schema_utils import webmethod

2
llama_toolchain/safety/client.py
View file

@ -9,7 +9,7 @@ import asyncio
import fire
import httpx
from llama_models.llama3_1.api.datatypes import UserMessage
from llama_models.llama3.api.datatypes import UserMessage
from termcolor import cprint
from .api import (

2
llama_toolchain/safety/meta_reference/shields/base.py
View file

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

2
llama_toolchain/safety/meta_reference/shields/contrib/third_party_shield.py
View file

@ -6,7 +6,7 @@
from typing import List
from llama_models.llama3_1.api.datatypes import Message
from llama_models.llama3.api.datatypes import Message
from llama_toolchain.safety.meta_reference.shields.base import (
OnViolationAction,

2
llama_toolchain/safety/meta_reference/shields/llama_guard.py
View file

@ -10,7 +10,7 @@ from string import Template
from typing import List, Optional
import torch
from llama_models.llama3_1.api.datatypes import Message, Role
from llama_models.llama3.api.datatypes import Message, Role
from transformers import AutoModelForCausalLM, AutoTokenizer
from .base import CANNED_RESPONSE_TEXT, OnViolationAction, ShieldBase, ShieldResponse

2
llama_toolchain/safety/meta_reference/shields/prompt_guard.py
View file

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