diff --git a/optimum_benchmark/backends/pytorch/config.py b/optimum_benchmark/backends/pytorch/config.py index 3efeb63c..d3ae3fbb 100644 --- a/optimum_benchmark/backends/pytorch/config.py +++ b/optimum_benchmark/backends/pytorch/config.py @@ -10,14 +10,7 @@ TORCH_DTYPES = ["bfloat16", "float16", "float32", "auto"] QUANTIZATION_CONFIGS = {"bnb": {"llm_int8_threshold": 0.0}, "gptq": {}, "awq": {}} -COMPILE_CONFIG = { - "fullgraph": False, - "dynamic": False, - "backend": "inductor", - "mode": None, - "options": None, - "disable": False, -} +COMPILE_CONFIG = {} @dataclass