Skip to content

Commit 0a979b3

Browse files
RobinPicardrlouf
authored andcommitted
Fix error with the tokonzer inferenace arg in Transformers
1 parent bc334e6 commit 0a979b3

File tree

3 files changed

+3
-2
lines changed

3 files changed

+3
-2
lines changed

outlines/models/transformers.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -220,7 +220,6 @@ def __init__(
220220

221221
tokenizer.padding_side = "left"
222222
self.model = model
223-
self.transformer_tokenizer = tokenizer
224223
self.tokenizer = TransformerTokenizer(tokenizer)
225224
self.type_adapter = TransformersTypeAdapter()
226225

@@ -332,9 +331,9 @@ def generate_stream(self, model_input, output_type, **inference_kwargs):
332331

333332
def _generate_output_seq(self, prompts, inputs, **inference_kwargs):
334333
input_ids = inputs["input_ids"]
334+
335335
output_ids = self.model.generate(
336336
**inputs,
337-
tokenizer=self.transformer_tokenizer,
338337
**inference_kwargs,
339338
)
340339

tests/v0_legacy/models/test_transformers_legacy.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -168,5 +168,6 @@ def test_transformers_legacy_call_generation():
168168
"foo",
169169
2,
170170
length_penalty=0.5,
171+
tokenizer=model.tokenizer.tokenizer,
171172
)
172173
assert isinstance(result, str)

tests/v0_legacy/models/test_transformers_vision_legacy.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -106,5 +106,6 @@ def test_transformers_vision_legacy_call_generation():
106106
"foo",
107107
2,
108108
length_penalty=0.5,
109+
tokenizer=model.tokenizer.tokenizer,
109110
)
110111
assert isinstance(result, str)

0 commit comments

Comments
 (0)