m.safari:

            ~~~~~~~~~~~~ <--- HERE



yeah you would need to modify the original model instead of wrapping it inside self.model_fp32 here I think. Or if you don’t have access to the original model, you could unpack the output of self.model_fp32(x) and Add a dequant stub for the ones that needs to be dequantized.