Skip to content

Commit ef6146b

Browse files
committed
Add tests for the Outlines model
1 parent 74cfc03 commit ef6146b

File tree

5 files changed

+629
-4
lines changed

5 files changed

+629
-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', token=None),
538+
AutoTokenizer.from_pretrained('erwanf/gpt2-mini', token=None),
539+
)
540+
)
529541
else:
530542
raise ValueError(f'Unknown model: {request.param}')
531543
except ImportError:

0 commit comments

Comments
 (0)