def METHOD_NAME(self):
    model = RegressionModel()
    model._original_forward = model.forward
    model.forward = convert_outputs_to_fp32(model.forward)
    model = extract_model_from_parallel(model, keep_fp32_wrapper=False)
    _ = pickle.dumps(model)