phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-10 07:35:59 +00:00
Code Issues Projects Releases Packages Wiki 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/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