fix import error in parallel build

[llan-ml]

When using custom model file and building the engine in parallel, the original implementation will raise import error.