Skip to content

Commit

Permalink
[WWB]: Added config to preprocessor call in VLMs (#1638)
Browse files Browse the repository at this point in the history
  • Loading branch information
AlexKoff88 authored Jan 30, 2025
1 parent ec50b5b commit 624eb00
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -118,7 +118,7 @@ def default_gen_answer(
preprocess_inputs = MODEL_TYPE_TO_CLS_MAPPING[
model.config.model_type
].preprocess_inputs
inputs = preprocess_inputs(prompt, image, processor, tokenizer)
inputs = preprocess_inputs(prompt, image, processor, tokenizer, config=model.config)
tokens = model.generate(
**inputs,
do_sample=False,
Expand Down

0 comments on commit 624eb00

Please sign in to comment.