Skip to content

Commit

Permalink
remove unnecessary
Browse files Browse the repository at this point in the history
  • Loading branch information
echarlaix committed Feb 18, 2025
1 parent 4582284 commit 0252eb2
Showing 1 changed file with 0 additions and 12 deletions.
12 changes: 0 additions & 12 deletions optimum/utils/input_generators.py
Original file line number Diff line number Diff line change
Expand Up @@ -1622,12 +1622,6 @@ def __init__(
self.height = preprocessor.crop_size.get("height", self.height)
self.width = preprocessor.crop_size.get("width", self.width)

def generate(self, input_name: str, framework: str = "pt", int_dtype: str = "int64", float_dtype: str = "fp32"):
input_ = super().generate(
input_name=input_name, framework=framework, int_dtype=int_dtype, float_dtype=float_dtype
)
return input_


class DummyVisionStaticInputGenerator(DummyVisionInputGenerator):
def __init__(
Expand Down Expand Up @@ -1657,12 +1651,6 @@ def __init__(
self.height = preprocessor.size.get("height", self.height)
self.width = preprocessor.size.get("width", self.width)

def generate(self, input_name: str, framework: str = "pt", int_dtype: str = "int64", float_dtype: str = "fp32"):
input_ = super().generate(
input_name=input_name, framework=framework, int_dtype=int_dtype, float_dtype=float_dtype
)
return input_


class PerceiverDummyInputGenerator(DummyVisionStaticInputGenerator):
pass
Expand Down

0 comments on commit 0252eb2

Please sign in to comment.