ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

huggingface/mistralai/Mistral-7B-Instruct-v0.3

Browse Source
This commit is contained in:
Ishaan Jaff 2025-01-13 18:42:36 -08:00
parent 3fe1f3b3b2
commit 970e9c7507
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/llm_translation/test_text_completion_unit_tests.py
View File

@ -102,7 +102,7 @@ async def test_huggingface_text_completion_logprobs():
client = AsyncHTTPHandler()
with patch.object(client, "post", return_value=return_val) as mock_post:
response = await litellm.atext_completion(
model="huggingface/mistralai/Mistral-7B-v0.1",
model="huggingface/mistralai/Mistral-7B-Instruct-v0.3",
prompt="good morning",
client=client,
)

4
tests/local_testing/test_text_completion.py
View File

@ -3940,7 +3940,7 @@ def test_completion_hf_prompt_array():
litellm.set_verbose = True
print("\n testing hf mistral\n")
response = text_completion(
model="huggingface/mistralai/Mistral-7B-v0.1",
model="huggingface/mistralai/Mistral-7B-Instruct-v0.3",
prompt=token_prompt, # token prompt is a 2d list,
max_tokens=0,
temperature=0.0,
@ -3971,7 +3971,7 @@ def test_text_completion_stream():
try:
for _ in range(2): # check if closed client used
response = text_completion(
model="huggingface/mistralai/Mistral-7B-v0.1",
model="huggingface/mistralai/Mistral-7B-Instruct-v0.3",
prompt="good morning",
stream=True,
max_tokens=10,