mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-07-09 23:25:58 +00:00
llama_models.llama3_1 -> llama_models.llama3
This commit is contained in:
parent
f502716cf7
commit
38244c3161
27 changed files with 44 additions and 42 deletions
|
@ -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
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue