Skip to content

Commit

Permalink
lint
Browse files Browse the repository at this point in the history
  • Loading branch information
jc9123 committed Jun 10, 2024
1 parent cbfe892 commit bff8ba7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/vision_language_model_example.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,13 +16,13 @@ def run_idefics2_pixel_values(*, disable_image_processor: bool = False):
image_token_id=32000,
image_input_shape="1,3,980,980",
image_feature_size=576,
disable_image_processor=disable_image_processor,
dtype='float16',
)

prompt = "<image>" * 64 + (
"\nUSER: What is the content of this image?\nASSISTANT:")

# This should be provided by another online or offline component.
if disable_image_processor:
image = torch.load("images/bird_pixel_values.pt")
else:
Expand Down

0 comments on commit bff8ba7

Please sign in to comment.