Skip to content

Commit b45cc86

Browse files
committed
Add tests for the Outlines model
1 parent 0762360 commit b45cc86

File tree

5 files changed

+831
-4
lines changed

5 files changed

+831
-4
lines changed

tests/conftest.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -536,6 +536,18 @@ def model(
536536
'Qwen/Qwen2.5-72B-Instruct',
537537
provider=HuggingFaceProvider(provider_name='nebius', api_key=huggingface_api_key),
538538
)
539+
elif request.param == 'outlines':
540+
from outlines.models.transformers import from_transformers
541+
from transformers import AutoModelForCausalLM, AutoTokenizer
542+
543+
from pydantic_ai.models.outlines import OutlinesModel
544+
545+
return OutlinesModel(
546+
from_transformers(
547+
AutoModelForCausalLM.from_pretrained('erwanf/gpt2-mini'),
548+
AutoTokenizer.from_pretrained('erwanf/gpt2-mini'),
549+
)
550+
)
539551
else:
540552
raise ValueError(f'Unknown model: {request.param}')
541553
except ImportError:

0 commit comments

Comments
 (0)