huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-04-24 00:12:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

wip(ci): rerun ci to debug

Browse Source
This commit is contained in:
baptiste 2025-04-10 11:47:40 +00:00
parent 9fdc67af5c
commit db98b4611b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
integration-tests/gaudi/test_gaudi_generate.py
View File

@ -3,7 +3,7 @@ from typing import Any, Dict
from text_generation import AsyncClient from text_generation import AsyncClient
import pytest import pytest
# The "args" config is not optimized for speed but only check that the inference is working for the different models architectures # The "args" values in TEST_CONFIGS are not optimized for speed but only check that the inference is working for the different models architectures.
TEST_CONFIGS = { TEST_CONFIGS = {
"meta-llama/Llama-3.1-8B-Instruct-shared": { "meta-llama/Llama-3.1-8B-Instruct-shared": {
"model_id": "meta-llama/Llama-3.1-8B-Instruct", "model_id": "meta-llama/Llama-3.1-8B-Instruct",