### Problem Description The current code replaces modules after AutoScheme, which may cause the layer_config generated by AutoScheme to be incompatible. ### Reproduction Steps ~ ### Environment Information _No response_ ### Error Logs ```shell ``` ### Additional Context _No response_