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

fixed imports models.llama3. --> models.llama3_1.api.

Browse source
This commit is contained in:
Hardik Shah 2024-07-19 17:42:14 -07:00
parent f94efcf2ee
commit 2ed2881a21
20 changed files with 26 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

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

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