Skip to content

Commit f7d6e88

Browse files
authored
Revert "Adding DummyInputGenerator for VisualBert" (#2306)
Revert "Adding DummyInputGenerator for VisualBert (#2303)" This reverts commit a4152c0.
1 parent eb187fa commit f7d6e88

File tree

1 file changed

+0
-33
lines changed

1 file changed

+0
-33
lines changed

optimum/utils/input_generators.py

Lines changed: 0 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -63,8 +63,6 @@ def wrapper(*args, **kwargs):
6363
"num_channels": 3,
6464
"point_batch_size": 3,
6565
"nb_points_per_image": 2,
66-
"visual_seq_length": 16,
67-
"visual_embedding_dim": 20,
6866
# audio
6967
"feature_size": 80,
7068
"nb_max_frames": 3000,
@@ -808,9 +806,6 @@ class DummyVisionInputGenerator(DummyInputGenerator):
808806
"pixel_mask",
809807
"sample",
810808
"latent_sample",
811-
"visual_embeds",
812-
"visual_token_type_ids",
813-
"visual_attention_mask",
814809
)
815810

816811
def __init__(
@@ -821,8 +816,6 @@ def __init__(
821816
num_channels: int = DEFAULT_DUMMY_SHAPES["num_channels"],
822817
width: int = DEFAULT_DUMMY_SHAPES["width"],
823818
height: int = DEFAULT_DUMMY_SHAPES["height"],
824-
visual_seq_length: int = DEFAULT_DUMMY_SHAPES["visual_seq_length"],
825-
visual_embedding_dim: int = DEFAULT_DUMMY_SHAPES["visual_embedding_dim"],
826819
**kwargs,
827820
):
828821
self.task = task
@@ -846,8 +839,6 @@ def __init__(
846839
self.image_size = (self.image_size, self.image_size)
847840
self.batch_size = batch_size
848841
self.height, self.width = self.image_size
849-
self.visual_seq_length = visual_seq_length
850-
self.visual_embedding_dim = visual_embedding_dim
851842

852843
def generate(self, input_name: str, framework: str = "pt", int_dtype: str = "int64", float_dtype: str = "fp32"):
853844
if input_name == "pixel_mask":
@@ -857,30 +848,6 @@ def generate(self, input_name: str, framework: str = "pt", int_dtype: str = "int
857848
framework=framework,
858849
dtype=int_dtype,
859850
)
860-
861-
elif input_name in "visual_attention_mask":
862-
return self.random_mask_tensor(
863-
shape=[self.batch_size, self.visual_seq_length],
864-
padding_side="right",
865-
framework=framework,
866-
dtype=int_dtype,
867-
)
868-
869-
elif input_name == "visual_token_type_ids":
870-
return self.random_int_tensor(
871-
shape=[self.batch_size, self.visual_seq_length],
872-
max_value=1,
873-
framework=framework,
874-
dtype=int_dtype,
875-
)
876-
877-
elif input_name == "visual_embeds":
878-
return self.random_float_tensor(
879-
shape=[self.batch_size, self.visual_seq_length, self.visual_embedding_dim],
880-
framework=framework,
881-
dtype=float_dtype,
882-
)
883-
884851
else:
885852
return self.random_float_tensor(
886853
shape=[self.batch_size, self.num_channels, self.height, self.width],

0 commit comments

Comments
 (0)