riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-08-02 03:01:07 +06:00
Code Issues Actions 3 Packages Projects Releases Wiki Activity

add bbox input validation (#26294)

Browse Source
This commit is contained in:
Jinho Park 2023-09-20 23:48:35 +09:00 committed by GitHub
parent 245532065d
commit 00247ea0de
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
src/transformers/models/bros/modeling_bros.py
View File

@ -877,6 +877,9 @@ class BrosModel(BrosPreTrainedModel):
else: else:
raise ValueError("You have to specify either input_ids or inputs_embeds") raise ValueError("You have to specify either input_ids or inputs_embeds")
if bbox is None:
raise ValueError("You have to specify bbox")
batch_size, seq_length = input_shape batch_size, seq_length = input_shape
device = input_ids.device if input_ids is not None else inputs_embeds.device device = input_ids.device if input_ids is not None else inputs_embeds.device
@ -924,13 +927,11 @@ class BrosModel(BrosPreTrainedModel):
past_key_values_length=past_key_values_length, past_key_values_length=past_key_values_length,
) )
bbox_position_embeddings = None # if bbox has 2 points (4 float tensors) per token, convert it to 4 points (8 float tensors) per token
if bbox is not None: if bbox.shape[-1] == 4:
# if bbox has 2 points (4 float tensors) per token, convert it to 4 points (8 float tensors) per token bbox = bbox[:, :, [0, 1, 2, 1, 2, 3, 0, 3]]
if bbox.shape[-1] == 4: scaled_bbox = bbox * self.config.bbox_scale
bbox = bbox[:, :, [0, 1, 2, 1, 2, 3, 0, 3]] bbox_position_embeddings = self.bbox_embeddings(scaled_bbox)
scaled_bbox = bbox * self.config.bbox_scale
bbox_position_embeddings = self.bbox_embeddings(scaled_bbox)
encoder_outputs = self.encoder( encoder_outputs = self.encoder(
embedding_output, embedding_output,