enforce AOBaseConfig type in quantize_'s config argument

[vkuzo]

Summary:

Finishing the migration in #1690. After this PR, AOBaseConfig is the only way to configure a workflow to be used in quantize_.

Before

def quantize_(
    model: torch.nn.Module,
    config: Union[AOBaseConfig, Callable[[torch.nn.Module], torch.nn.Module]],
    filter_fn: Optional[Callable[[torch.nn.Module, str], bool]] = None,
    set_inductor_config: Optional[bool] = None,
    device: Optional[torch.types.Device] = None,
):

After

def quantize_(
    model: torch.nn.Module,
    config: AOBaseConfig,
    filter_fn: Optional[Callable[[torch.nn.Module, str], bool]] = None,
    set_inductor_config: Optional[bool] = None,
    device: Optional[torch.types.Device] = None,
):

Test Plan: CI

Reviewers:

Subscribers:

Tasks:

Tags: