mirror of
https://github.com/huggingface/transformers.git
synced 2025-08-03 03:31:05 +06:00
fix: Update pixel_values parameter in hf_model input (#34782)
This commit is contained in:
parent
e80a65ba4f
commit
1ef6c5f1c5
@ -249,7 +249,7 @@ def convert_blip2_checkpoint(
|
||||
{"image": original_pixel_values, "text_input": [caption]}, match_head="itm"
|
||||
)
|
||||
logits = hf_model(
|
||||
pixel_values=original_pixel_values,
|
||||
pixel_values=pixel_values,
|
||||
input_ids=input_ids,
|
||||
attention_mask=attention_mask,
|
||||
use_image_text_matching_head=True,
|
||||
@ -274,7 +274,7 @@ def convert_blip2_checkpoint(
|
||||
{"image": original_pixel_values, "text_input": [caption]}, match_head="itc"
|
||||
)
|
||||
logits = hf_model(
|
||||
pixel_values=original_pixel_values,
|
||||
pixel_values=pixel_values,
|
||||
input_ids=input_ids,
|
||||
attention_mask=attention_mask,
|
||||
use_image_text_matching_head=False,
|
||||
|
Loading…
Reference in New Issue
Block a user