prompt formatting for together ai llama2 models

This commit is contained in:
Krrish Dholakia 2023-09-05 11:57:13 -07:00
parent dc130efac6
commit 64f3d3c56e
4 changed files with 31 additions and 23 deletions

View file

@ -7,7 +7,7 @@ import time
from typing import Callable
from litellm.utils import ModelResponse
from typing import Optional
from .huggingface_model_prompt_templates.factory import prompt_factory, custom_prompt
from .prompt_templates.factory import prompt_factory, custom_prompt
class HuggingfaceError(Exception):
def __init__(self, status_code, message):