def METHOD_NAME(fp32_model, dummy_input, quant_scheme):
    sim = QuantizationSimModel(fp32_model,
                               dummy_input,
                               quant_scheme=quant_scheme)
    sim.compute_encodings(lambda model, _: model(dummy_input), None)
    return sim.model