Skip to content

Commit 528e5ea

Browse files
committed
Add tests for the Outlines model
1 parent 4e8ebc1 commit 528e5ea

File tree

5 files changed

+715
-4
lines changed

5 files changed

+715
-4
lines changed

tests/conftest.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -544,6 +544,18 @@ def model(
544544
'Qwen/Qwen2.5-72B-Instruct',
545545
provider=HuggingFaceProvider(provider_name='nebius', api_key=huggingface_api_key),
546546
)
547+
elif request.param == 'outlines':
548+
from outlines.models.transformers import from_transformers
549+
from transformers import AutoModelForCausalLM, AutoTokenizer
550+
551+
from pydantic_ai.models.outlines import OutlinesModel
552+
553+
return OutlinesModel(
554+
from_transformers(
555+
AutoModelForCausalLM.from_pretrained('erwanf/gpt2-mini'),
556+
AutoTokenizer.from_pretrained('erwanf/gpt2-mini'),
557+
)
558+
)
547559
else:
548560
raise ValueError(f'Unknown model: {request.param}')
549561
except ImportError:

0 commit comments

Comments
 (0)