Skip to content

Commit 7582e1e

Browse files
committed
Add tests for the Outlines model
1 parent 3c92a23 commit 7582e1e

File tree

6 files changed

+626
-4
lines changed

6 files changed

+626
-4
lines changed

tests/conftest.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -526,6 +526,18 @@ def model(
526526
'Qwen/Qwen2.5-72B-Instruct',
527527
provider=HuggingFaceProvider(provider_name='nebius', api_key=huggingface_api_key),
528528
)
529+
elif request.param == 'outlines':
530+
from outlines.models.transformers import from_transformers
531+
from transformers import AutoModelForCausalLM, AutoTokenizer
532+
533+
from pydantic_ai.models.outlines import OutlinesModel
534+
535+
return OutlinesModel(
536+
from_transformers(
537+
AutoModelForCausalLM.from_pretrained('erwanf/gpt2-mini'),
538+
AutoTokenizer.from_pretrained('erwanf/gpt2-mini'),
539+
)
540+
)
529541
else:
530542
raise ValueError(f'Unknown model: {request.param}')
531543
except ImportError:

0 commit comments

Comments
 (0)